def main(): # set path iris_path = [ 'DataAnalysisProjectDesign/Experiment1/iris_train.arff', 'DataAnalysisProjectDesign/Experiment1/iris_test.arff' ] adult_path = [ 'DataAnalysisProjectDesign/Experiment1/adult_train.arff', 'DataAnalysisProjectDesign/Experiment1/adult_test.arff' ] # get choice data_choice = input('Enter 1 for iris DT; Enter 2 for adult DT:') tree_choice = input('Enter 1 for ID3; Enter 2 for CART:') path = select_dataset(data_choice, iris_path, adult_path) # create train data instance train_data_obj = Data(path[0]) train_data_obj.load_data() train_data_obj.fill_missing_data() # create test data instance test_data_obj = Data(path[1]) test_data_obj.clear_memory() test_data_obj.load_data() test_data_obj.fill_missing_data() tree = dt_router(train_data_obj, test_data_obj, tree_choice) tree.test() conf_mat, judge = tree.get_conf_mat() return tree, conf_mat, judge
def main(): # set path iris_path = 'DataAnalysisProjectDesign/Experiment2/iris_train.arff' adult_path = 'DataAnalysisProjectDesign/Experiment2/adult_train.arff' # get choice data_choice = input('Enter 1 for iris; Enter 2 for adult:') dt_num = int(input('Enter your expected tree number:')) path = select_dataset(data_choice,iris_path,adult_path) # create data instance data_obj = Data(path) data_obj.load_data() data_obj.fill_missing_data() # create random forest rf = RandomForest( data=data_obj, dt_num=dt_num ) rf.bagging() rf.train_rf() correct_rate,conf_mat = rf.test_rf() return dt_num,correct_rate,conf_mat