Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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 :)
4 changes: 2 additions & 2 deletions pystacknet/pystacknet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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

Expand Down