python-scikit-learnHow to get decision tree classifier accuracy score
from sklearn import datasets, tree, model_selection, metrics
X, y = datasets.load_iris(return_X_y=True)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, y)
model = tree.DecisionTreeClassifier()
model.fit(X_train, y_train)
accuracy_score = metrics.accuracy_score(model.predict(X_test), y_test)ctrl + cfrom sklearn importimport module from scikit-learn | load_irisloads Iris dataset |
model_selection.train_test_splitsplits given | .DecisionTreeClassifier(creates decision tree classifier model |
.fit(train model with a given features and target variable dataset | .predict(predict target variable based on given features dataset |
.accuracy_score(accuracy classification score | |
Usage example
from sklearn import datasets, tree, model_selection, metrics
X, y = datasets.load_iris(return_X_y=True)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, y)
model = tree.DecisionTreeClassifier()
model.fit(X_train, y_train)
accuracy_score = metrics.accuracy_score(model.predict(X_test), y_test)
print(accuracy_score)output
0.9210526315789473
Related
More of Python Scikit Learn
- TSNe embedding example
- SVR regression example
- Using resample example
- How to inverse transform PCA reduced data
- Decision tree regression example
- Decision tree classifier example
- Get linear regression model coefficients
- How to get best parameters from grid search CV
- Using quantile transformer example
- PCA dimensionality reduction example
See more codes...