Попытка перейти к регрессии #1
This commit is contained in:
6
main.py
6
main.py
@@ -4,16 +4,16 @@ import visual
|
||||
import neuro_defs
|
||||
|
||||
|
||||
dataset = generate.generate_dataset(10_000)
|
||||
dataset = generate.generate_dataset(1000)
|
||||
|
||||
|
||||
# Создаём и обучаем сеть
|
||||
nn = neuro_defs.SimpleNN()
|
||||
nn.train(dataset.train, epochs=10)
|
||||
nn.train(dataset.train, dataset.train_answs, epochs=100)
|
||||
|
||||
# Проверяем на новой точке
|
||||
for dot in dataset.test[:10]:
|
||||
print(nn.forward(dot.x, dot.y), dot.__repr__())
|
||||
print(nn.forward(dot), dot)
|
||||
|
||||
# visual.plot_dataset(dataset)
|
||||
# visual.plt_show()
|
||||
Reference in New Issue
Block a user