machine-learning-notebooks icon indicating copy to clipboard operation
machine-learning-notebooks copied to clipboard

Support for best_estimator_?

Open browshanravan opened this issue 4 years ago • 0 comments

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?

browshanravan avatar Dec 24 '20 17:12 browshanravan