make_prediction_dlhub¶
- mastml.mastml_predictor.make_prediction_dlhub(input_dict)[source]¶
Prediction script, same functionality as make_prediction above, but tailored for model running on DLHub/Foundry
Use this function as the function pointer for DLHub PythonStaticMethodModel (see ‘Foundry_model_upload_example.ipynb’ in main mastml folder)
- Things that need to be uploaded:
model.pkl (must have this name) X_train.xlsx (or X_train.csv) (must have this name)
- Optional to upload:
preprocessor.pkl (must have this name)
- Args:
- input_dict (dict): dictionary containing at least the following: {‘X_test’: pd.DataFrame() of featurized test data}
The keys are the input arguments of mastml_predictor (see above for explanation)