CubeBit/StreetsignRecognition
1
0
Fork 0
mirror of https://github.com/3x4byte/StreetsignRecognition.git synced 2025-12-21 01:15:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: nn learn now takes training data in its order (this prevents that some vectors are never used)

Browse Source
This commit is contained in:
Lukas Karras
2024-11-26 08:34:27 +01:00
parent 2b1b183247
commit c724954f78
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/classes/neural/network.py
View File

@@ -38,7 +38,7 @@ class NeuralNetwork:
correct_amount = 0 correct_amount = 0
false_amount = 0 false_amount = 0
for i in range(len(training_set)): 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) classified_concept, correct = self.classify(fv_to_train)
if not correct: if not correct:

4
src/main.py
View File

@@ -68,9 +68,9 @@ def neural_network(training = os.path.abspath(os.path.join(__file__, "..", "trai
next(reader) next(reader)
i = 0 i = 0
for row in reader: for row in reader:
i += 1
if i > num_training_vectors: if i > num_training_vectors:
break break
i += 1
float_list = ast.literal_eval(row[3]) 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 return [cnt_correct, cnt_data - cnt_correct, cnt_data] # richtig, falsch, amount of testing data
if __name__ == "__main__": if __name__ == "__main__":
for i in range(100, 10000, 100):
print(f"{i} training vectors: ", end="")
neural_network() neural_network()