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

Improving textDatasetGenerator control - renaming source files

parent e8179f0c
Loading
Loading
Loading
Loading
+24 −0
Original line number Diff line number Diff line
import numpy as np
import math
import random
from textDatasetGenerator import generateTextDataset
import sys

def additional_input(k, x, y):
    if k == 0:
@@ -88,6 +90,10 @@ class Dataset:
            self.createSpiral()
            return True
        
        elif(type == "custom"):
            self.createCustomDatasetFromFiles()
            return True
        
        else:
            return False
        
@@ -199,3 +205,21 @@ class Dataset:
                self.labels[i] = 1
        self.labels = 2*self.labels-1

    def createCustomDatasetFromFiles(self):
        if not generateTextDataset(self.data_size):
            print("Datasize for custom dataset must be even number.")
            sys.exit()
        self.labelled_data_A = np.load('custom_datasets/classA.npy')
        self.labelled_data_B = np.load('custom_datasets/classB.npy')
        self.all_data = np.zeros((self.data_size, self.input_coord_dimension))
        print(self.data_size)
        self.labels = np.zeros(self.data_size)
        
        for i in range(len(self.labelled_data_A)):
            self.labels[i] = 1
            self.all_data[i] = self.labelled_data_A[i]
        
        for i in range(len(self.labelled_data_B)):
            self.all_data[i+len(self.labelled_data_A)] = self.labelled_data_B[i]

        self.labels = 2*self.labels-1
+0 −0

File moved.

−69.2 KiB (528 KiB)
Loading image diff...
+57.6 KiB (1.81 MiB)
Loading image diff...
+333 −100

File changed.

Preview size limit exceeded, changes collapsed.

Loading