示例#1
0
print("RuleSet:\n", RuleSet.head)
print("\n\n")

# - Part B - Rules Mining ------------------------------------------------------------------------------------------------

# - Part C - Predictive Analytics ------------------------------------------------------------------------------------------------
print("- Part C - Predictive Analytics ------------------------------------------------------------------------------------------------")
# Decision Tree
print("Decision Tree on balance-scale dataset:")

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