예제 #1
0
    def __init__(self, layer_sizes=[16, 2000, 4], lr=.1, activation=ANN.relu, max_iterations=10, rand_limit_min=-.02, rand_limit_max=.02,
                 learningSet = [], learningSet_answ = [], testSet = [], testSet_answ = []):

        # Converting data...
        print("Converting data...")
        learningSet = self.convert_input_divide_relative_to_max(learningSet)
        learningSet_answ = self.convert_answers(learningSet_answ)
        testSet = self.convert_input_divide_relative_to_max(testSet)

        # Builds the network
        self.neuralNet = ANN.neuralnetwork(layer_sizes=layer_sizes, lr=lr, activation=activation, max_iterations=max_iterations, rand_limit_min=rand_limit_min,
                                           rand_limit_max=rand_limit_max, learningSet=learningSet, learningSet_answ=learningSet_answ, testSet=testSet, testSet_answ=testSet_answ)