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,
|