Первый прототип нейросети готов

classes.py

@@ -1,18 +1,18 @@
 import random
 
 class DataSet:
-	def __init__(self, N=1000) -> None:
+	def __init__(self, func, N=1000) -> None:
 		self.train = []
 		self.train_answs = []
 		self.test = []
 		self.test_answs = []
 
 		for i in range(N//5*4):
-			x = random.uniform(-1000, 1000)
+			x = random.uniform(1, 9)
 			self.train.append(x)
-			self.train_answs.append(x+1)
+			self.train_answs.append(func(x))
 
 		for i in range(N//5):
-			x = random.uniform(-1000, 1000)
+			x = random.uniform(1, 9)
 			self.test.append(x)
-			self.test_answs.append(x+1)
+			self.test_answs.append(func(x))