Commit 734d1a21 authored by Stenli Karanxha's avatar Stenli Karanxha
Browse files

Enabled the full testing of the main script.

parent ace194e1
......@@ -36,7 +36,7 @@ OUTPUT_FILE_SEPARATOR = ';'
CLASSES_NUMBER = 87
TRAIN_SAMPLES_NUMBER = 100 #687
TRAIN_SAMPLES_NUMBER = 687
TEST_SAMPLES_NUMBER = 1000
PERCENT_FOR_LEARNING = 75 # % of the training set used for the learning. The rest goes to the verification.
......@@ -308,7 +308,7 @@ if __name__ == '__main__':
prepare_learn_data() and prepare_verification_data():
#print_all_data()
#run_algorithm(Dummy(), DUMMY_OUT_FILE_NAME, 'Dummy')
run_algorithm(Dummy(), DUMMY_OUT_FILE_NAME, 'Dummy')
run_algorithm(LinearRegression(num_features, CLASSES_NUMBER), LINEAR_REGRESSION_OUT_FILE_NAME, 'Linear Regression')
#un_algorithm(NeuralNetwork(num_features,30,CLASSES_NUMBER), NEURAL_NETWORKS_OUT_FILE_NAME, 'Neural Network')
#run_algorithm(KNearestNeighbour(4), K_NEAREST_NEIGHBOURS_OUT_FILE_NAME, 'K-nearest neighbours')
un_algorithm(NeuralNetwork(num_features,30,CLASSES_NUMBER), NEURAL_NETWORKS_OUT_FILE_NAME, 'Neural Network')
run_algorithm(KNearestNeighbour(4), K_NEAREST_NEIGHBOURS_OUT_FILE_NAME, 'K-nearest neighbours')
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment