fedex/mt
Public Access
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

results ae section

Browse Source
This commit is contained in:
Jan Kowalczyk
2025-09-18 11:58:28 +02:00
parent 8f36bd2e07
commit a20a4a0832
4 changed files with 393 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tools/plot_scripts/load_results.py
View File

@@ -75,7 +75,6 @@ PRETRAIN_SCHEMA = {
"semi_anomalous": pl.Int32,
"model": pl.Utf8, # always "ae"
"fold": pl.Int32,
"split": pl.Utf8, # "train" | "test"
# timings and optimization
"train_time": pl.Float64,
"test_time": pl.Float64,
@@ -577,7 +576,7 @@ def load_pretraining_results_dataframe(
# Cast/optimize a bit (categoricals, ints, floats)
df = df.with_columns(
pl.col("network", "model", "split").cast(pl.Categorical),
pl.col("network", "model").cast(pl.Categorical),
pl.col(
"latent_dim", "semi_normals", "semi_anomalous", "fold", "k_fold_num"
).cast(pl.Int32),