Open pplonski opened 3 years ago
I'll like to work on this
@asuzukosi great!
Some tips:
predict()
method. Model name will be simply a string that is displayed in the report.best_model
are loaded (for speed reason). You need to ensure that model selected for prediction is loaded. In the case of Ensemble that might be several models.Please let me know if you need more help.
hey, is this issue still open? can i work on this
hey @matrixhead!
the issue is still open, I'm assigning it to you
I don't know this is the right way to do it
@matrixhead few comments:
model_name
.load_on_predict
parameter in params.json
file are loaded. Those are models needed be best_model. We need to assure that selected model by user is loaded.predict()
, predict_proba()
and predict_all()
methods. They should support the model_name
parameter.Hi @pplonski , I just opened a PR for this. I'm sure that more work would be required before we can merge it, but early feedback would be very helpful. Thanks!
Quick and dirty workaround until this is oficially implemented:
from supervised.model_framework import ModelFramework
model = ModelFramework.load("AUTOML_FOLDER_NAME_HERE", model_subpath="MODEL_FOLDER_NAME_HERE")
For regression results:
y_pred = automl._base_predict(qdqd_X_test, model=model)["prediction"].to_numpy()
For classification results:
y_pred = automl._base_predict(qdqd_X_test, model=model)["label"].to_numpy()
Hopefully i can save someones time. :D
Quick and dirty workaround until this is oficially implemented:
from supervised.model_framework import ModelFramework model = ModelFramework.load("AUTOML_FOLDER_NAME_HERE", model_subpath="MODEL_FOLDER_NAME_HERE")
For regression results:
y_pred = automl._base_predict(qdqd_X_test, model=model)["prediction"].to_numpy()
For classification results:
y_pred = automl._base_predict(qdqd_X_test, model=model)["label"].to_numpy()
Hopefully i can save someones time. :D
I am trying to implement this work around, but I am getting the error "module 'supervised.automl' has no attribute '_base_predict'". Do I need to be loading automl in a different way than "from supervised import automl"?
Thank you @Reese-Martin, during summer we will have intern, I hope this will be implemented :)
Details in discussion https://github.com/mljar/mljar-supervised/discussions/421