Skip to content

Commit 4d40cd6

Browse files
committed
fix model stats JSON export in compile_model_stats.py and use model-stats-uniq-protos.json on landing page to match metrics table
update rolling-mae-vs-hull-dist-wbm-batches-*.svelte + add gnome variant add MACE + Alignn rolling MAE per batch to preprint/+page.md
1 parent 7e549e2 commit 4d40cd6

18 files changed

+39
-20
lines changed

matbench_discovery/plots.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -537,10 +537,14 @@ def rolling_mae_vs_hull_dist(
537537

538538
from matbench_discovery.preds import model_styles
539539

540-
for trace in fig.data:
540+
for idx, trace in enumerate(fig.data):
541541
if style := model_styles.get(trace.name):
542542
ls, _marker, color = style
543543
trace.line = dict(color=color, dash=ls, width=2)
544+
else: # pick from default colors, line styles, markers
545+
trace.line = dict(
546+
color=plotly_colors[idx], dash=plotly_line_styles[idx], width=2
547+
)
544548
# marker_spacing = 2
545549
# fig.add_scatter(
546550
# x=trace.x[::marker_spacing],

scripts/model_figs/compile_model_stats.py

+13-4
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,13 @@
1919
from tqdm import tqdm
2020

2121
from matbench_discovery import PDF_FIGS, SITE_FIGS, SITE_LIB, WANDB_PATH
22-
from matbench_discovery.preds import df_metrics, df_metrics_10k, df_preds, model_styles
22+
from matbench_discovery.preds import (
23+
df_metrics,
24+
df_metrics_10k,
25+
df_metrics_uniq_protos,
26+
df_preds,
27+
model_styles,
28+
)
2329

2430
__author__ = "Janosh Riebesell"
2531
__date__ = "2022-11-28"
@@ -107,10 +113,14 @@
107113

108114

109115
# %%
110-
for df_tmp, label in ((df_metrics, ""), (df_metrics_10k, "-10k")):
116+
for df_tmp, label in (
117+
(df_metrics, ""),
118+
(df_metrics_10k, "-10k"),
119+
(df_metrics_uniq_protos, "-uniq-protos"),
120+
):
111121
df_tmp = pd.concat(
112122
[
113-
df_metrics,
123+
df_tmp,
114124
pd.DataFrame(train_stats).add_prefix("Train ", axis="index"),
115125
pd.DataFrame(test_stats).add_prefix("Test ", axis="index"),
116126
],
@@ -125,7 +135,6 @@
125135
]
126136

127137
df_tmp.attrs["All Models Run Time"] = df_tmp[time_col].sum()
128-
print(f"{df_tmp[time_col].sum()=:.0f} hours")
129138

130139
df_tmp.round(2).to_json(f"{SITE_LIB}/model-stats{label}.json", orient="index")
131140
if label == "":

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-alignn.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-bowsr.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-cgcnn+p.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-cgcnn.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-chgnet.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-gnome.svelte

+1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-m3gnet.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-mace.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-megnet.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-voronoi-rf.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/figs/rolling-mae-vs-hull-dist-wbm-batches-wrenformer.svelte

+1-1
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

site/src/lib/model-stats-10k.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
{"GNoME":{"F1":0.81,"DAF":4.84,"Precision":0.82,"Recall":0.8,"Accuracy":0.94,"TPR":0.8,"FPR":0.04,"TNR":0.96,"FNR":0.2,"TP":35082.0,"FP":7422.0,"TN":203981.0,"FN":8736.0,"MAE":0.03,"RMSE":0.08,"R2":0.78,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":1742,"missing_percent":"0.68%"},"MACE":{"F1":0.67,"DAF":3.5,"Precision":0.58,"Recall":0.78,"Accuracy":0.87,"TPR":0.78,"FPR":0.12,"TNR":0.88,"FNR":0.22,"TP":33428.0,"FP":23943.0,"TN":182497.0,"FN":9360.0,"MAE":0.06,"RMSE":0.1,"R2":0.66,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":111.9,"Test GPU":4.0,"Test CPU":64.0,"Test Slurm Jobs":100.0,"Run Time (h)":111.9,"missing_preds":7735,"missing_percent":"3.01%"},"CHGNet":{"F1":0.61,"DAF":3.04,"Precision":0.52,"Recall":0.74,"Accuracy":0.84,"TPR":0.74,"FPR":0.14,"TNR":0.86,"FNR":0.26,"TP":32642.0,"FP":29980.0,"TN":182891.0,"FN":11450.0,"MAE":0.06,"RMSE":0.1,"R2":0.68,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":142.48,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":100.0,"Run Time (h)":142.48,"missing_preds":0,"missing_percent":"0.00%"},"M3GNet":{"F1":0.58,"DAF":2.65,"Precision":0.45,"Recall":0.79,"Accuracy":0.8,"TPR":0.79,"FPR":0.2,"TNR":0.8,"FNR":0.21,"TP":34731.0,"FP":41738.0,"TN":170780.0,"FN":9361.0,"MAE":0.07,"RMSE":0.12,"R2":0.58,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":83.65,"Test GPU":0.0,"Test CPU":76.0,"Test Slurm Jobs":99.0,"Run Time (h)":83.65,"missing_preds":353,"missing_percent":"0.14%"},"ALIGNN":{"F1":0.56,"DAF":2.92,"Precision":0.5,"Recall":0.65,"Accuracy":0.83,"TPR":0.65,"FPR":0.13,"TNR":0.87,"FNR":0.35,"TP":28598.0,"FP":28465.0,"TN":184406.0,"FN":15494.0,"MAE":0.09,"RMSE":0.15,"R2":0.27,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":0,"missing_percent":"0.00%"},"MEGNet":{"F1":0.51,"DAF":2.7,"Precision":0.46,"Recall":0.57,"Accuracy":0.81,"TPR":0.57,"FPR":0.14,"TNR":0.86,"FNR":0.43,"TP":25311.0,"FP":29343.0,"TN":183528.0,"FN":18781.0,"MAE":0.13,"RMSE":0.2,"R2":-0.28,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":3.44,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":1.0,"Run Time (h)":3.44,"missing_preds":0,"missing_percent":"0.00%"},"CGCNN":{"F1":0.51,"DAF":2.63,"Precision":0.45,"Recall":0.59,"Accuracy":0.81,"TPR":0.59,"FPR":0.15,"TNR":0.85,"FNR":0.41,"TP":25895.0,"FP":31475.0,"TN":181394.0,"FN":18197.0,"MAE":0.14,"RMSE":0.23,"R2":-0.62,"Train Run Time (h)":11.35,"Train GPU":1.0,"Train CPU":128.0,"Train Slurm Jobs":10.0,"Test Run Time (h)":0.17,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":1.0,"Run Time (h)":11.52,"missing_preds":2,"missing_percent":"0.00%"},"CGCNN+P":{"F1":0.51,"DAF":2.4,"Precision":0.41,"Recall":0.67,"Accuracy":0.78,"TPR":0.67,"FPR":0.2,"TNR":0.8,"FNR":0.33,"TP":29557.0,"FP":42282.0,"TN":170587.0,"FN":14535.0,"MAE":0.11,"RMSE":0.18,"R2":0.03,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":2,"missing_percent":"0.00%"},"Wrenformer":{"F1":0.48,"DAF":2.13,"Precision":0.36,"Recall":0.69,"Accuracy":0.74,"TPR":0.69,"FPR":0.25,"TNR":0.75,"FNR":0.31,"TP":30570.0,"FP":53088.0,"TN":159783.0,"FN":13522.0,"MAE":0.1,"RMSE":0.18,"R2":-0.04,"Train Run Time (h)":57.89,"Train GPU":1.0,"Train CPU":128.0,"Train Slurm Jobs":10.0,"Test Run Time (h)":0.45,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":1.0,"Run Time (h)":58.34,"missing_preds":0,"missing_percent":"0.00%"},"BOWSR":{"F1":0.44,"DAF":1.91,"Precision":0.32,"Recall":0.74,"Accuracy":0.68,"TPR":0.74,"FPR":0.33,"TNR":0.67,"FNR":0.26,"TP":31347.0,"FP":68140.0,"TN":140149.0,"FN":11143.0,"MAE":0.12,"RMSE":0.16,"R2":0.14,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":2705.45,"Test GPU":0.0,"Test CPU":32.0,"Test Slurm Jobs":476.0,"Run Time (h)":2705.45,"missing_preds":6184,"missing_percent":"2.41%"},"Voronoi RF":{"F1":0.34,"DAF":1.51,"Precision":0.26,"Recall":0.51,"Accuracy":0.66,"TPR":0.51,"FPR":0.3,"TNR":0.7,"FNR":0.49,"TP":22517.0,"FP":64431.0,"TN":148424.0,"FN":21574.0,"MAE":0.14,"RMSE":0.21,"R2":-0.31,"Train Run Time (h)":173.78,"Train GPU":0.0,"Train CPU":76.0,"Train Slurm Jobs":48.0,"Test Run Time (h)":30.92,"Test GPU":0.0,"Test CPU":76.0,"Test Slurm Jobs":20.0,"Run Time (h)":204.7,"missing_preds":17,"missing_percent":"0.01%"}}
1+
{"GNoME":{"F1":0.97,"DAF":5.62,"Precision":0.94,"Recall":1.0,"Accuracy":0.94,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":9371.0,"FP":629.0,"TN":0.0,"FN":0.0,"MAE":0.04,"RMSE":0.09,"R2":0.86,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":1742,"missing_percent":"0.68%"},"CHGNet":{"F1":0.94,"DAF":5.29,"Precision":0.88,"Recall":1.0,"Accuracy":0.88,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":8817.0,"FP":1183.0,"TN":0.0,"FN":0.0,"MAE":0.06,"RMSE":0.11,"R2":0.81,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":142.48,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":100.0,"Run Time (h)":142.48,"missing_preds":0,"missing_percent":"0.00%"},"MACE":{"F1":0.9,"DAF":4.88,"Precision":0.81,"Recall":1.0,"Accuracy":0.81,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":8125.0,"FP":1875.0,"TN":0.0,"FN":0.0,"MAE":0.09,"RMSE":0.23,"R2":0.22,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":7735,"missing_percent":"3.01%"},"M3GNet":{"F1":0.89,"DAF":4.79,"Precision":0.8,"Recall":1.0,"Accuracy":0.8,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":7980.0,"FP":2020.0,"TN":0.0,"FN":0.0,"MAE":0.1,"RMSE":0.16,"R2":0.62,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":353,"missing_percent":"0.14%"},"CGCNN":{"F1":0.76,"DAF":3.66,"Precision":0.61,"Recall":1.0,"Accuracy":0.61,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":6102.0,"FP":3898.0,"TN":0.0,"FN":0.0,"MAE":0.17,"RMSE":0.24,"R2":0.27,"Train Run Time (h)":11.35,"Train GPU":1.0,"Train CPU":128.0,"Train Slurm Jobs":10.0,"Test Run Time (h)":0.17,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":1.0,"Run Time (h)":11.52,"missing_preds":2,"missing_percent":"0.00%"},"ALIGNN":{"F1":0.75,"DAF":3.58,"Precision":0.6,"Recall":1.0,"Accuracy":0.6,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":5965.0,"FP":4035.0,"TN":0.0,"FN":0.0,"MAE":0.19,"RMSE":0.26,"R2":0.14,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":0,"missing_percent":"0.00%"},"CGCNN+P":{"F1":0.74,"DAF":3.53,"Precision":0.59,"Recall":1.0,"Accuracy":0.59,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":5887.0,"FP":4113.0,"TN":0.0,"FN":0.0,"MAE":0.21,"RMSE":0.29,"R2":0.03,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":2,"missing_percent":"0.00%"},"Wrenformer":{"F1":0.74,"DAF":3.52,"Precision":0.59,"Recall":1.0,"Accuracy":0.59,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":5873.0,"FP":4127.0,"TN":0.0,"FN":0.0,"MAE":0.19,"RMSE":0.25,"R2":0.22,"Train Run Time (h)":57.89,"Train GPU":1.0,"Train CPU":128.0,"Train Slurm Jobs":10.0,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":57.89,"missing_preds":0,"missing_percent":"0.00%"},"BOWSR":{"F1":0.69,"DAF":3.14,"Precision":0.52,"Recall":1.0,"Accuracy":0.52,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":5227.0,"FP":4773.0,"TN":0.0,"FN":0.0,"MAE":0.26,"RMSE":0.33,"R2":-1.1,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":2705.45,"Test GPU":0.0,"Test CPU":32.0,"Test Slurm Jobs":476.0,"Run Time (h)":2705.45,"missing_preds":6184,"missing_percent":"2.41%"},"MEGNet":{"F1":0.64,"DAF":2.84,"Precision":0.47,"Recall":1.0,"Accuracy":0.47,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":4727.0,"FP":5273.0,"TN":0.0,"FN":0.0,"MAE":0.33,"RMSE":0.36,"R2":-0.82,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":0,"missing_percent":"0.00%"},"Voronoi RF":{"F1":0.58,"DAF":2.45,"Precision":0.41,"Recall":1.0,"Accuracy":0.41,"TPR":1.0,"FPR":1.0,"TNR":0.0,"FNR":0.0,"TP":4082.0,"FP":5918.0,"TN":0.0,"FN":0.0,"MAE":0.36,"RMSE":0.43,"R2":-0.81,"Train Run Time (h)":173.78,"Train GPU":0.0,"Train CPU":76.0,"Train Slurm Jobs":48.0,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":173.78,"missing_preds":17,"missing_percent":"0.01%"}}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
{"GNoME":{"F1":0.83,"DAF":5.49,"Precision":0.84,"Recall":0.82,"Accuracy":0.95,"TPR":0.82,"FPR":0.03,"TNR":0.97,"FNR":0.18,"TP":27178.0,"FP":5009.0,"TN":175827.0,"FN":5957.0,"MAE":0.04,"RMSE":0.08,"R2":0.78,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":1742,"missing_percent":"0.68%"},"MACE":{"F1":0.67,"DAF":3.84,"Precision":0.58,"Recall":0.8,"Accuracy":0.88,"TPR":0.8,"FPR":0.11,"TNR":0.89,"FNR":0.2,"TP":25734.0,"FP":18924.0,"TN":157508.0,"FN":6560.0,"MAE":0.06,"RMSE":0.11,"R2":0.66,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":7735,"missing_percent":"3.01%"},"CHGNet":{"F1":0.61,"DAF":3.32,"Precision":0.51,"Recall":0.76,"Accuracy":0.85,"TPR":0.76,"FPR":0.13,"TNR":0.87,"FNR":0.24,"TP":25313.0,"FP":23955.0,"TN":158159.0,"FN":8061.0,"MAE":0.06,"RMSE":0.1,"R2":0.69,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":142.48,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":100.0,"Run Time (h)":142.48,"missing_preds":0,"missing_percent":"0.00%"},"M3GNet":{"F1":0.57,"DAF":2.84,"Precision":0.44,"Recall":0.8,"Accuracy":0.81,"TPR":0.8,"FPR":0.19,"TNR":0.81,"FNR":0.2,"TP":26797.0,"FP":34034.0,"TN":147781.0,"FN":6577.0,"MAE":0.08,"RMSE":0.12,"R2":0.58,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":353,"missing_percent":"0.14%"},"ALIGNN":{"F1":0.57,"DAF":3.16,"Precision":0.49,"Recall":0.67,"Accuracy":0.84,"TPR":0.67,"FPR":0.13,"TNR":0.87,"FNR":0.33,"TP":22436.0,"FP":23346.0,"TN":158768.0,"FN":10938.0,"MAE":0.09,"RMSE":0.15,"R2":0.3,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":0,"missing_percent":"0.00%"},"MEGNet":{"F1":0.51,"DAF":2.92,"Precision":0.45,"Recall":0.58,"Accuracy":0.83,"TPR":0.58,"FPR":0.13,"TNR":0.87,"FNR":0.42,"TP":19537.0,"FP":23651.0,"TN":158463.0,"FN":13837.0,"MAE":0.13,"RMSE":0.21,"R2":-0.25,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":0,"missing_percent":"0.00%"},"CGCNN":{"F1":0.51,"DAF":2.82,"Precision":0.44,"Recall":0.6,"Accuracy":0.82,"TPR":0.6,"FPR":0.14,"TNR":0.86,"FNR":0.4,"TP":20191.0,"FP":26074.0,"TN":156038.0,"FN":13183.0,"MAE":0.14,"RMSE":0.23,"R2":-0.6,"Train Run Time (h)":11.35,"Train GPU":1.0,"Train CPU":128.0,"Train Slurm Jobs":10.0,"Test Run Time (h)":0.17,"Test GPU":1.0,"Test CPU":128.0,"Test Slurm Jobs":1.0,"Run Time (h)":11.52,"missing_preds":2,"missing_percent":"0.00%"},"CGCNN+P":{"F1":0.5,"DAF":2.53,"Precision":0.39,"Recall":0.69,"Accuracy":0.79,"TPR":0.69,"FPR":0.2,"TNR":0.8,"FNR":0.31,"TP":23117.0,"FP":35893.0,"TN":146219.0,"FN":10257.0,"MAE":0.11,"RMSE":0.18,"R2":0.02,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":0.0,"missing_preds":2,"missing_percent":"0.00%"},"Wrenformer":{"F1":0.47,"DAF":2.23,"Precision":0.34,"Recall":0.72,"Accuracy":0.74,"TPR":0.72,"FPR":0.25,"TNR":0.75,"FNR":0.28,"TP":23990.0,"FP":45586.0,"TN":136528.0,"FN":9384.0,"MAE":0.11,"RMSE":0.19,"R2":-0.04,"Train Run Time (h)":57.89,"Train GPU":1.0,"Train CPU":128.0,"Train Slurm Jobs":10.0,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":57.89,"missing_preds":0,"missing_percent":"0.00%"},"BOWSR":{"F1":0.43,"DAF":2.0,"Precision":0.3,"Recall":0.74,"Accuracy":0.7,"TPR":0.74,"FPR":0.31,"TNR":0.69,"FNR":0.26,"TP":23963.0,"FP":55843.0,"TN":122841.0,"FN":8357.0,"MAE":0.12,"RMSE":0.17,"R2":0.15,"Train Run Time (h)":null,"Train GPU":null,"Train CPU":null,"Train Slurm Jobs":null,"Test Run Time (h)":2705.45,"Test GPU":0.0,"Test CPU":32.0,"Test Slurm Jobs":476.0,"Run Time (h)":2705.45,"missing_preds":6184,"missing_percent":"2.41%"},"Voronoi RF":{"F1":0.33,"DAF":1.56,"Precision":0.24,"Recall":0.54,"Accuracy":0.67,"TPR":0.54,"FPR":0.31,"TNR":0.69,"FNR":0.46,"TP":17854.0,"FP":56122.0,"TN":125990.0,"FN":15520.0,"MAE":0.15,"RMSE":0.21,"R2":-0.33,"Train Run Time (h)":173.78,"Train GPU":0.0,"Train CPU":76.0,"Train Slurm Jobs":48.0,"Test Run Time (h)":null,"Test GPU":null,"Test CPU":null,"Test Slurm Jobs":null,"Run Time (h)":173.78,"missing_preds":17,"missing_percent":"0.01%"}}

0 commit comments

Comments
 (0)