Skip to content

Commit 1681c38

Browse files
committed
merge load_model_preds() into load_df_wbm_with_preds() and pass through **kwargs to pandas reader
fix test_load_preds.py and test_plots.py
1 parent 44d5133 commit 1681c38

12 files changed

+254944
-32
lines changed

.gitignore

-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ job-logs/
2020

2121
# slurm logs
2222
*slurm-*.log
23-
models/**/*.csv
2423

2524
# temporary ignore rules
2625
paper

models/bowsr/join_bowsr_results.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535

3636

3737
# %%
38-
df_bowsr = pd.concat(dfs.values()).round(6)
38+
df_bowsr = pd.concat(dfs.values()).round(4)
3939

4040

4141
# %% compare against WBM formation energy targets to make sure we got sensible results

models/cgcnn/test_cgcnn.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@
110110
)
111111

112112
slurm_job_id = os.environ.get("SLURM_JOB_ID", "debug")
113-
df_preds.to_csv(f"{out_dir}/{job_name}-preds-{slurm_job_id}.csv", index=False)
113+
df_preds.round(4).to_csv(f"{out_dir}/{job_name}-preds-{slurm_job_id}.csv", index=False)
114114
pred_col = f"{target_col}_pred_ens"
115115
assert pred_col in df, f"{pred_col=} not in {list(df)}"
116116
table = wandb.Table(dataframe=df_preds[[target_col, pred_col]].reset_index())

0 commit comments

Comments
 (0)