Improving textDatasetGenerator control - renaming source files

import numpy as np

import math

import random

from textDatasetGenerator import generateTextDataset

import sys

def additional_input(k, x, y):

    if k == 0:

            self.createSpiral()

            return True

        elif(type == "custom"):

            self.createCustomDatasetFromFiles()

            return True

        else:

            return False

                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