Commit 679270d8 authored by Jan Kovář's avatar Jan Kovář
Browse files

Defined dataset type

parent 33da8bda
Loading
Loading
Loading
Loading
+5 −3
Original line number Diff line number Diff line
@@ -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)