Loading train.py +5 −3 Original line number Diff line number Diff line Loading @@ -12,14 +12,16 @@ validation_data_size = 250 #Number of test datapoints (test data is the final check, if your neural network is trained well - you should not optimize NN based on test data) test_data_size = data_size - training_data_size - validation_data_size #Dataset creation - there are several predefined datasets - look at dataset.py to method createAllData() #Dataset initialization data = Dataset(data_size, training_data_size, validation_data_size, test_data_size) if(data.initializeDataSet("spiral") == False): #Predefined datasets - look at dataset.py to method createAllData() dataset_type = "spiral" if(data.initializeDataSet(dataset_type) == False): print("Dataset not defined") sys.exit(1) #The structure is an array of number of neurons at corresponding layer, length of structure is the depth of neural network structure = [data.input_data_dimension,8,8,8,data.output_data_dimension] structure = [data.input_data_dimension,5,5,5,data.output_data_dimension] #Initialization of new neural network network = NeuralNetwork(structure) Loading Loading
train.py +5 −3 Original line number Diff line number Diff line Loading @@ -12,14 +12,16 @@ validation_data_size = 250 #Number of test datapoints (test data is the final check, if your neural network is trained well - you should not optimize NN based on test data) test_data_size = data_size - training_data_size - validation_data_size #Dataset creation - there are several predefined datasets - look at dataset.py to method createAllData() #Dataset initialization data = Dataset(data_size, training_data_size, validation_data_size, test_data_size) if(data.initializeDataSet("spiral") == False): #Predefined datasets - look at dataset.py to method createAllData() dataset_type = "spiral" if(data.initializeDataSet(dataset_type) == False): print("Dataset not defined") sys.exit(1) #The structure is an array of number of neurons at corresponding layer, length of structure is the depth of neural network structure = [data.input_data_dimension,8,8,8,data.output_data_dimension] structure = [data.input_data_dimension,5,5,5,data.output_data_dimension] #Initialization of new neural network network = NeuralNetwork(structure) Loading