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

decision tree implementation in python github

Aggressive Angelfish answered on August 19, 2023 Popularity 5/10 Helpfulness 1/10

answer decision tree implementation in python github

decision tree implementation in python github

Comment

# This is decision tree classifier for classification problem
# For regression, you need to search regression tree
# Determine independent and dependent features 
X= df.drop(['target'], axis=1)
y = df["target"]

# Transform categories into numeric values of second independent feature (column) 
from sklearn import preprocessing
encoder = preprocessing.LabelEncoder()
encoder.fit(['category 1','category 2','category 3'])
X[:,1] = encoder.transform(X[:,1])

# Split into train and test set
from sklearn.model_selection import train_test_split
X_Train, X_Test, y_Train, y_Test = train_test_split(X, y, test_size=0.3, random_state=3)

# Make sure to take into account the class imbalance 
from sklearn.utils.class_weight import compute_sample_weight
w_train = compute_sample_weight('balanced', y_train)

# Train the classifier
from sklearn.tree import DecisionTreeClassifier
tree_clf = DecisionTreeClassifier(criterion="entropy", max_depth = 4)
tree_clf.fit(X_Train,y_Train, sample_weight=w_train)

# Alternative approach : Train the classifier with snapml (offers multi-threaded CPU/GPU training)
from snapml import DecisionTreeClassifier
snapml_dt_gpu = DecisionTreeClassifier(max_depth=4, random_state=45, use_gpu=True)
snapml_dt_cpu = DecisionTreeClassifier(max_depth=4, random_state=45, n_jobs=4)
snapml_dt.fit(X_train, y_train, sample_weight=w_train)
# Predict
y_pred = tree_clf.predict(X_Test)

# Determine accuracy
from sklearn import metrics
metrics.accuracy_score(y_testset, predTree)

# Compute predicted probabilities
y_pred_prob = tree_clf.predict_proba(X_test)[:,1]

# Evaluate tree
from sklearn.metrics import roc_auc_score
sklearn_roc_auc = roc_auc_score(y_test, y_pred)

# Visualize the tree
from sklearn import tree
tree.plot_tree(tree_clf)
plt.show()

Popularity 5/10 Helpfulness 1/10 Language python

Source: Grepper

Tags: decision-tree github implementation python

Link to this answer
Share Copy Link

Contributed on Aug 19 2023

Aggressive Angelfish

0 Answers Avg Quality 2/10

decision tree implementation in python github

Contents

More Related Answers

decision tree implementation in python github

Grepper

Documentation

Social

Legal

Contact

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