Spaces:

tschouis
/

tox21_snn_classifier

Sleeping

antoniaebner commited on 12 days ago

Commit

9ccf171

1 Parent(s): c33530b

bug fix

Files changed (1) hide show

predict.py CHANGED Viewed

@@ -22,7 +22,7 @@ ECDFS_PATH = "data/ecdfs.pkl"
 FEATURE_SELECTION_PATH = "data/feat_selection.npz"
 SCALER1_PATH = "data/scaler1.pkl"
 SCALER2_PATH = "data/scaler2.pkl"
-CHECKPOINT_PATH = "checkpoints/noble-butterfly-499_last.pth.pth"
 ECFP_RADIUS = 3
 ECFP_FPSIZE = 8192

 FEATURE_SELECTION_PATH = "data/feat_selection.npz"
 SCALER1_PATH = "data/scaler1.pkl"
 SCALER2_PATH = "data/scaler2.pkl"
+CHECKPOINT_PATH = "checkpoints/noble-butterfly-499_last.pth"
 ECFP_RADIUS = 3
 ECFP_FPSIZE = 8192