diff --git a/README.md b/README.md index cab00f8..a4b3758 100644 --- a/README.md +++ b/README.md @@ -88,3 +88,7 @@ random_state | Integer for randomised procedures n_jobs | Number of models to run in parallel. This is independent of any extra threads allocated n_jobs | Number of models to run in parallel. This is independent of any extra threads allocated from the selected algorithms. e.g. it is possible to run 4 models in parallel where one is a randomforest that runs on 10 threads (it selected). verbose | Integer value higher than zero to allow printing at the console. + + ------------------------------------------------------------------------------------------------- + I fixed it for sklearn 0.23.0 + now It works greatly :) diff --git a/pystacknet/pystacknet.py b/pystacknet/pystacknet.py index 356f77e..d824053 100644 --- a/pystacknet/pystacknet.py +++ b/pystacknet/pystacknet.py @@ -19,7 +19,7 @@ from sklearn.model_selection import KFold from sklearn.utils import check_X_y,check_array,check_consistent_length, column_or_1d import inspect -from sklearn.externals.joblib import delayed,Parallel +from joblib import delayed,Parallel import operator import time from sklearn.preprocessing import LabelEncoder @@ -97,7 +97,7 @@ def _parallel_predict_proba_scoring(estimators, X, index): raise Exception (" predictions' shape not equal among estimators within the batch as %d!=%d " % (predictions.shape[1],preds.shape[1])) preds+=predictions - preds/=float(len(estimators)) + preds= np.true_divide(preds,float(len(estimators))) return preds,index