import pandas as pd

import numpy as np
from sklearn import metrics
from sklearn import tree

################################################

train=pd.read_csv("Entrenamiento.csv");
train.head();
print("DATASET DE ENTRENAMIENTO:\n", train);

################################################

x_train = np.array(train.drop(['Class'], 1))

y_train = np.array(train['Class'])
print(x_train.shape)

################################################

test=pd.read_csv("Prueba.csv");
test.head();
print("DATASET DE PRUEBA:\n", test);

################################################

# VALORES DE PRUEBA
x_test = np.array(train.drop(['Class'], 1))
y_test = np.array(train['Class'])
print(x_test.shape)

################################################

from sklearn.tree import DecisionTreeClassifier

classifier = DecisionTreeClassifier(criterion = 'entropy', max_depth = 5,
random_state = 1)
classifier.fit(x_train, y_train)
y_pred = classifier.predict(x_test)

################################################

cm = confusion_matrix(y_test, y_pred)
print('MATRIZ DE CONFUSION PARA EL MODELO ARBOL DE DECISION\n ')
print('',cm)

################################################

presicion=metrics.precision_score(y_test, y_pred, average='macro')

print('Precision:', presicion)
recall= metrics.recall_score(y_test, y_pred,average='macro')
print('Recall:',recall)
print("Accuracy:",metrics.accuracy_score(y_test, y_pred))
print('F-Measure: ', 2*((presicion*recall)/(presicion+recall)))
exactitud = accuracy_score(y_test, y_pred)
print('Exactitud:',exactitud)

################################################

print(metrics.classification_report(y_test, y_pred))