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

integrate iterator in Stree

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-05-19 18:19:23 +02:00
parent 95a6901f47
commit 6ebd0f9be3
5 changed files with 53 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@@ -48,6 +48,7 @@ print(clf)
print(f"Classifier's accuracy (train): {clf.score(Xtrain, ytrain):.4f}")
print(f"Classifier's accuracy (test) : {clf.score(Xtest, ytest):.4f}")
proba = clf.predict_proba(Xtest)
print("Checking that we have correct probabilities, these are probabilities of sample belonging to class 1")
res0 = proba[proba[:, 0] == 0]
res1 = proba[proba[:, 0] == 0]
print("++++++++++res0++++++++++++")