예제 #1
0
Dataset_Path = 'balance-scale.csv'

print("\n\n")

# Building Phase
Dataset = dt.ImportDataset(Dataset_Path)
X, Y, X_train, X_test, Y_train, Y_test = dt.SplitDataset(Dataset)
DT_Gini = dt.Train_Gini(X_train, Y_train)
DT_Entropy = dt.Train_Entropy(X_train, Y_train)

print("\n\n")
    
# Operational Phase
# Prediction using Gini
print("Results Using Gini Index:")
Y_pred_Gini = dt.Predict(X_test, DT_Gini)
dt.PrintAccuracy(Y_test, Y_pred_Gini)

print("\n\n")

# Prediction using Entropy
print("Results Using Entropy:")
Y_pred_Entropy = dt.Predict(X_test, DT_Entropy)
dt.PrintAccuracy(Y_test, Y_pred_Entropy)

print("\n\n")

# Bayes Classifier
print("\n\n")
print("Bayes Classifier on Iris dataset:")