源码:
文档:
写一个入口代码进行调试
from sklearn.datasets import load_iris
from sklearn.metrics import accuracy_score
from sklearn.model_selection import train_test_split
from numpy_ml.trees import DecisionTree
X, y = load_iris(True)
dt = DecisionTree()
X_train, X_test, y_train, y_test = train_test_split(X, y)
dt.fit(X_train, y_train)
y_pred = dt.predict(X_test)
score = accuracy_score