manifoldai / merf

Mixed Effects Random Forest
MIT License
219 stars 52 forks source link

method predict gives values higher than 1 #57

Open veseliy opened 4 years ago

veseliy commented 4 years ago

Here is the merf model, that gives prob higher than one zeroes_zf_m.pickle.zip

X = np.array([[0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 1.]])
with open('zeroes_zf_m.pickle', 'rb') as f:
    merf_model = pkl.load(f)

print(model.predict(X, Z=[[1]], clusters=pd.Series(42)), model.predict(X, Z=[[1]], clusters=pd.Series('42')))

Is it correct? My opinion that probs must be [0,1]