machine-learning-notebooks
machine-learning-notebooks copied to clipboard
Support for best_estimator_?
I was wondering if you have any plans to include the learned best_estimator_
attribute for your EstimatorSelectionHelper()
function?
This will allow using the calculated best estimator and its corresponding hyper-parameters to be used for test data evaluation and more.
I was hoping to achieve something like this:
helper = EstimatorSelectionHelper(models, params)
helper.fit(X_train, y_train, scoring='roc_auc')
helper.best_estimator_.predict(X_test)
I guess given that people might like to select a particular parameter such as "mean_score" as measure of best estimator, it needs to be provided as an option somewhere. Perhaps as an argument in the fit process?