Update src/prediction_pipeline.py
Browse files
src/prediction_pipeline.py
CHANGED
@@ -69,11 +69,6 @@ class ActivityPredictor:
|
|
69 |
support_inactives_smiles
|
70 |
)
|
71 |
|
72 |
-
# save inputs
|
73 |
-
import pickle
|
74 |
-
with open("/system/user/publicwork/luukkonen/mhnfs-benchmark/js_code/preprocess_data/ap_inputs.pkl", "wb") as f:
|
75 |
-
pickle.dump((query_input, support_actives_input, support_inactives_input, support_actives_size, support_inactives_size), f)
|
76 |
-
|
77 |
# Make predictions
|
78 |
predictions = self.model(
|
79 |
query_input,
|
|
|
69 |
support_inactives_smiles
|
70 |
)
|
71 |
|
|
|
|
|
|
|
|
|
|
|
72 |
# Make predictions
|
73 |
predictions = self.model(
|
74 |
query_input,
|