python-scikit-learnDecision tree regression example

from sklearn import datasets, tree, model_selection

X, y = datasets.load_diabetes(return_X_y=True)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, y)

model = tree.DecisionTreeRegressor()
model.fit(X_train, y_train)

score = model.score(X_test, y_test)ctrl + cgithub

`from sklearn import` import module from scikit-learn	`datasets.load_diabetes` loads sample diabetes database
`model_selection.train_test_split` splits given `X` and `y` datasets to test (25% of values by default) and train (75% of values by default) subsets	`.DecisionTreeRegressor(` creates decision tree regression model
`.fit(` train model with a given features and target variable dataset	`.score(` returns model accuracy score

Usage example

from sklearn import datasets, tree, model_selection

X, y = datasets.load_diabetes(return_X_y=True)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, y)

model = tree.DecisionTreeRegressor()
model.fit(X_train, y_train)

print(model.score(X_test, y_test))

output

-0.1321271764882921

More of Python Scikit Learn

See more codes...

python-scikit-learnDecision tree regression example

Usage example

output

Related

More of Python Scikit Learn