comparison_tree.py

from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score

import pandas as pd
import numpy as np
import time
import pprint

from data_importation import importation
from print_time import print_time

########################
### Data Importation ###
########################

path = 'data/data.txt'
X,y = importation(path)
X_train, X_test, y_train, y_test = train_test_split(X, y,test_size = 0.25, random_state = 4)

######################
### Model Training ###
######################

print("Training...")
tree = DecisionTreeClassifier(max_depth = 10, min_samples_leaf = 4)
start = time.time()
tree.fit(X_train, y_train)
end = time.time()
print("Training time:")
print_time(start,end)

#####################
### Model Testing ###
#####################

print("Testing...")
start = time.time()
y_pred = tree.predict(X_test)
end = time.time()
print("Testing time:")
print_time(start,end)

################
### Accuracy ###
################

score = accuracy_score(y_test, y_pred)
print("Accuracy score:")
print(score)