此代码给我这个错误:
self.model = create_model((name error:name
'create_model'未定义
from sklearn.neural_network import MLPRegressor
class model:
def __init__(self, algorithm):
self.algorithm = algorithm
self.model = create_model()
def create_model(self):
if self.algorithm == 'MLPREG':
return MLPRegressor(hidden_layer_sizes=(20,), activation='relu', solver='adam', random_state=1, max_iter=3000, verbose=True)
def main():
ex = model('MLPREG')
main()
我不明白,一切看起来正常。但是我无法运行该程序。
将您的类更改为:
class model:
def __init__(self, algorithm):
self.algorithm = algorithm
self.model = self.create_model()
def create_model(self):
if self.algorithm == 'MLPREG':
return MLPRegressor(hidden_layer_sizes=(20,), activation='relu', solver='adam', random_state=1, max_iter=3000, verbose=True)