mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-15 23:46:02 +00:00
Working tree with samples and first test
This commit is contained in:
10
main.py
Normal file
10
main.py
Normal file
@@ -0,0 +1,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)
|
||||
model = Stree(random_state=random_state)
|
||||
model.fit(X, y)
|
||||
model.show_outcomes()
|
Reference in New Issue
Block a user