contoled data set prediction model

This commit is contained in:
aliaksei-kudravets 2022-05-26 11:00:43 +02:00
parent c06d68e0c3
commit 1bda1ab232

View File

@ -9,7 +9,7 @@ from pybrain3.tools.xml.networkwriter import NetworkWriter
from pybrain3.tools.xml.networkreader import NetworkReader
# https://www.machinelearningmastery.ru/how-to-configure-the-number-of-layers-and-nodes-in-a-neural-network/
class BackPropagationModel():
class SupervisedDataSetModel():
def __init__(self, metrics:int = 4,
predictions:int = 1,
input_layer:int = 4,
@ -54,7 +54,7 @@ class BackPropagationModel():
fileObject.close()
return net2
model = BackPropagationModel()
model = SupervisedDataSetModel()
model.addDataToModel([2, 3, 80, 1], [5])
model.addDataToModel([5, 5, 50, 2], [4])