diff --git a/main.py b/main.py
index 546c399..d33a9bc 100644
--- a/main.py
+++ b/main.py
@@ -4,7 +4,7 @@ import visual
 import neuro_defs
 
 
-dataset = generate.generate_dataset(100)
+dataset = generate.generate_dataset(1000)
 
 
 # Создаём и обучаем сеть
diff --git a/neuro_defs.py b/neuro_defs.py
index f82edb2..5b9fff7 100644
--- a/neuro_defs.py
+++ b/neuro_defs.py
@@ -47,7 +47,7 @@ class SimpleNN:
 		# self.w2_out = auto_diff.Node(random.uniform(-1, 1))
 		# self.b_out = auto_diff.Node(random.uniform(-1, 1))
 
-		self.network = classes.NeuronNetwork(4, 2, 2)
+		self.network = classes.NeuronNetwork(4, 1, 2)
 
 		self.lr = 0.1   # скорость обучения