首页 > 解决方案 > 如何解决问题类函数未定义?

问题描述

这段代码给了我这个错误:

self.model = create_model() NameError: 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()

我不明白,一切看起来都很正常。但我无法运行该程序。

标签: pythonfunctionclass

解决方案


将您的班级更改为:

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)

推荐阅读