Breaking News: Grepper is joining You.com. Read the official announcement!

decision tree regression python

Innocent Iguana answered on November 13, 2022 Popularity 9/10 Helpfulness 5/10

answer decision tree regression python

related decision tree regression

decision tree regression python

Comment

X = df.drop(columns=["target"])
Y = df["target"]
# Split into training and testing set
from sklearn.model_selection import train_test_split
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=.2, random_state=1)

# Model
from sklearn.tree import DecisionTreeRegressor
regression_tree = DecisionTreeRegressor(criterion = "mse")
regression_tree.fit(X_train, Y_train)

# Alternative approach
# in contrast to sklearn's Decision Tree, Snap ML offers multi-threaded CPU/GPU training 
from snapml import DecisionTreeRegressor
snapml_dt_gpu = DecisionTreeRegressor(max_depth=4, random_state=45, use_gpu=True)
snapml_dt_cpu = DecisionTreeRegressor(max_depth=8, random_state=45, n_jobs=4)
snapml_dt.fit(X_train, y_train)

# Predict
y_pred = regression_tree.predict(X_test)

# Evaluate the model
from sklearn.metrics import mean_squared_error
regression_tree.score(X_test, Y_test) # R-squared
mean_squared_error(y_test, y_pred) # MSE

Popularity 9/10 Helpfulness 5/10 Language python

Source: Grepper

Tags: decision-tree python regression

Link to this answer
Share Copy Link

Contributed on Feb 12 2023

Innocent Iguana

0 Answers Avg Quality 2/10

Closely Related Answers

decision tree regression

Comment

# Import the necessary modules and libraries
import numpy as np
from sklearn.tree import DecisionTreeRegressor
import matplotlib.pyplot as plt

# Create a random dataset
rng = np.random.RandomState(1)
X = np.sort(5 * rng.rand(80, 1), axis=0)
y = np.sin(X).ravel()
y[::5] += 3 * (0.5 - rng.rand(16))

# Fit regression model
regr_1 = DecisionTreeRegressor(max_depth=2)
regr_2 = DecisionTreeRegressor(max_depth=5)
regr_1.fit(X, y)
regr_2.fit(X, y)

# Predict
X_test = np.arange(0.0, 5.0, 0.01)[:, np.newaxis]
y_1 = regr_1.predict(X_test)
y_2 = regr_2.predict(X_test)

# Plot the results
plt.figure()
plt.scatter(X, y, s=20, edgecolor="black", c="darkorange", label="data")
plt.plot(X_test, y_1, color="cornflowerblue", label="max_depth=2", linewidth=2)
plt.plot(X_test, y_2, color="yellowgreen", label="max_depth=5", linewidth=2)
plt.xlabel("data")
plt.ylabel("target")
plt.title("Decision Tree Regression")
plt.legend()
plt.show()

Popularity 10/10 Helpfulness 3/10 Language whatever

Source: scikit-learn.org

Tags: decision-tree de

Link to this answer
Share Copy Link

Contributed on Nov 13 2022

Duangkamon Sawetwet

0 Answers Avg Quality 2/10

decision tree regression python

Contents

More Related Answers

decision tree regression python

Closely Related Answers

decision tree regression

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.