Как я могу решить проблему, функция класса не определена?

Question

Этот код дает мне эту ошибку:

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

Я не понимаю, все выглядит нормально.Но я не могу запустить программу.

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)