Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-19 01:16:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

test results & fix single predict_proba

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-05-18 17:08:05 +02:00
parent 382ae921ab
commit 68512b3d75
4 changed files with 113 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@@ -46,4 +46,12 @@ clf.fit(Xtrain, ytrain)
print(f"Took {time.time() - now:.2f} seconds to train")
print(clf)
print(f"Classifier's accuracy (train): {clf.score(Xtrain, ytrain):.4f}")
print(f"Classifier's accuracy (test) : {clf.score(Xtest, ytest):.4f}")
print(f"Classifier's accuracy (test) : {clf.score(Xtest, ytest):.4f}")
proba = clf.predict_proba(Xtest)
res0 = proba[proba[:, 0] == 0]
res1 = proba[proba[:, 0] == 0]
print("++++++++++res0++++++++++++")
print(res0[res0[:, 1] > .8])
print("**********res1************")
print(res1[res1[:, 1] < .4])
print(clf.predict_proba(Xtest))