Commit 2d757e47 authored by Jan Kovář's avatar Jan Kovář
Browse files

Final spell check

parent 679270d8

dataset.py

+3 −3
Original line number Diff line number Diff line
@@ -79,11 +79,11 @@ class Dataset: 
            self.createLinear()

            return True 



        elif(type == "elipse"):

        elif(type == "ellipse"):

            self.createElipse()

            return True



        elif(type == "two_elipses"):

        elif(type == "two_ellipses"):

            self.createTwoElipses()

            return True
@@ -91,7 +91,7 @@ class Dataset: 
            self.createSinus()

            return True



        elif(type == "half_elipse"):

        elif(type == "half_ellipse"):

            self.createHalfElipse()

            return True

textDatasetGenerator.py

+2 −2
Original line number Diff line number Diff line
@@ -15,10 +15,10 @@ def generateTextDataset(data_size): 
    np.random.seed(42)



    # Type chosen text

    text = "+"

    text = "D"



    # Add the custom text size

    font_size = 1000

    font_size = 900



    # Change path to your chosen font

    font_path = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"

train.py

+3 −3
Original line number Diff line number Diff line
@@ -6,16 +6,16 @@ import sys 
#Generating dataset - total number of datapoints

data_size = 1000

#Number of training datapoints

training_data_size = 500

training_data_size = 600

#Number of validation datapoints (used for hyperparameters optimization - changing dimensions of neural network, loss function, optimization method, learning rate,...)

validation_data_size = 250

validation_data_size = 200

#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 initialization

data = Dataset(data_size, training_data_size, validation_data_size, test_data_size)

#Predefined datasets - look at dataset.py to method createAllData()

dataset_type = "spiral"

dataset_type = "linear" #two_ellipses, half_ellipse, sinus, spiral

if(data.initializeDataSet(dataset_type) == False):

    print("Dataset not defined")

    sys.exit(1)