Learn practical skills, build real-world projects, and advance your career

Decision Tree

Classification Tree

import numpy as np 
import pandas as pd 
import sklearn
data=pd.read_csv('datasets/wine.csv')
data.head()

from sklearn import preprocessing 
  
# label_encoder object knows how to understand word labels. 
label_encoder = preprocessing.LabelEncoder() 
  
# Encode labels in column 'Wine Type'. 
data['Wine Type']= label_encoder.fit_transform(data['Wine Type'])