Updated 5 years ago
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'])