如何解决类函数未定义的问题？

Question

此代码显示以下错误：

self.model = create_model() NameError: name

'create_model' is not defined

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()

我不明白，一切正常。但我不能运行程序。你知道吗

Answer 1

将班级改为：

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)