From c724954f784fdd48c7ea6e1f9ebbed42e3a45daf Mon Sep 17 00:00:00 2001 From: Lukas Karras Date: Tue, 26 Nov 2024 08:34:27 +0100 Subject: [PATCH] fix: nn learn now takes training data in its order (this prevents that some vectors are never used) --- src/classes/neural/network.py | 2 +- src/main.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/classes/neural/network.py b/src/classes/neural/network.py index d1f8236..153bade 100644 --- a/src/classes/neural/network.py +++ b/src/classes/neural/network.py @@ -38,7 +38,7 @@ class NeuralNetwork: correct_amount = 0 false_amount = 0 for i in range(len(training_set)): - fv_to_train = random.choice(training_set) + fv_to_train = training_set # random.choice(training_set) classified_concept, correct = self.classify(fv_to_train) if not correct: diff --git a/src/main.py b/src/main.py index d208407..4f89992 100644 --- a/src/main.py +++ b/src/main.py @@ -68,9 +68,9 @@ def neural_network(training = os.path.abspath(os.path.join(__file__, "..", "trai next(reader) i = 0 for row in reader: - i += 1 if i > num_training_vectors: break + i += 1 float_list = ast.literal_eval(row[3]) @@ -100,4 +100,6 @@ def neural_network(training = os.path.abspath(os.path.join(__file__, "..", "trai return [cnt_correct, cnt_data - cnt_correct, cnt_data] # richtig, falsch, amount of testing data if __name__ == "__main__": - neural_network() \ No newline at end of file + for i in range(100, 10000, 100): + print(f"{i} training vectors: ", end="") + neural_network() \ No newline at end of file