contoled data set prediction model
This commit is contained in:
parent
c06d68e0c3
commit
1bda1ab232
@ -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])
|
||||
|
Loading…
Reference in New Issue
Block a user