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