mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-16 07:56:06 +00:00
Ricardo Montañana
f1ee4de37b
rename get_dataset to load_dataset add features and impurity to __str__ of node
21 lines
585 B
Python
21 lines
585 B
Python
import time
|
|
from sklearn.model_selection import train_test_split
|
|
from sklearn.datasets import load_iris
|
|
from stree import Stree
|
|
|
|
random_state = 1
|
|
|
|
X, y = load_iris(return_X_y=True)
|
|
|
|
Xtrain, Xtest, ytrain, ytest = train_test_split(
|
|
X, y, test_size=0.2, random_state=random_state
|
|
)
|
|
|
|
now = time.time()
|
|
clf = Stree(C=0.01, random_state=random_state)
|
|
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}")
|