This commit is contained in:
Denys Seredenko
2024-11-26 16:52:53 +01:00
parent 61ab3810e2
commit f09861a01c
4 changed files with 48 additions and 45 deletions

View File

@@ -33,11 +33,11 @@ class NeuralNetwork:
return [interpritated, True]
# takes feature vectors to train and generates weights
def learn(self, training_set: list[FeatureVector]):
def learn(self, training_set: list[FeatureVector], epochs = 10):
correct_amount = 0
false_amount = 0
for i in range(len(training_set)*100):
for i in range(len(training_set)*epochs):
fv_to_train = training_set[i%len(training_set)] # random.choice(training_set)
classified_concept, correct = self.classify(fv_to_train)