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

compute predictor and store model in node

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-05-13 00:12:05 +02:00
parent 371257c121
commit c4de782a3f
8 changed files with 263 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@@ -2,10 +2,10 @@ from trees.Stree import Stree
from sklearn.datasets import make_classification
random_state = 1
X, y = make_classification(n_samples=1500, n_features=3, n_informative=3,
n_redundant=0, n_repeated=0, n_classes=2, n_clusters_per_class=2,
class_sep=1.5, flip_y=0,weights=[0.5,0.5], random_state=random_state)
X, y = make_classification(n_samples=1500, n_features=3, n_informative=3,
n_redundant=0, n_repeated=0, n_classes=2, n_clusters_per_class=2,
class_sep=1.5, flip_y=0, weights=[0.5, 0.5], random_state=random_state)
model = Stree(random_state=random_state)
model.fit(X, y)
model.show_outcomes()
model.save_sub_datasets()
print(model)
model.save_sub_datasets()