Как я могу решить эту ошибку? NameError: имя 'модель' не определено

1

Когда я пытаюсь ввести текст, чтобы сделать предсказания, выполнение дает мне модель "NameError: name" не определена "

оценить модель нейронной сети

def evaluate_mode(Xtrain, ytrain, Xtest, ytest):

    scores = list()
    n_repeats = 2
    n_words = Xtest.shape[1]
    for i in range(n_repeats):
        # define network
        model = Sequential()
        model.add(Dense(50, input_shape=(n_words,), activation='relu'))
        model.add(Dense(1, activation='sigmoid'))
        # compile network
        model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
        # fit network
        model.fit(Xtrain, ytrain, epochs=10, verbose=2)
        # evaluate
        loss, acc = model.evaluate(Xtest, ytest, verbose=0)
        scores.append(acc)

        print('%d accuracy: %s' % ((i+1), acc))
    return scores

подготовить пакет слов, кодирующих документы

def prepare_data(train_docs, test_docs, mode):

    # create the tokenizer
    tokenizer = Tokenizer()
    # fit the tokenizer on the documents
    tokenizer.fit_on_texts(train_docs)
    # encode training data set
    Xtrain = tokenizer.texts_to_matrix(train_docs, mode=mode)
    # encode testing data set
    Xtest = tokenizer.texts_to_matrix(test_docs, mode=mode)
    return Xtrain, Xtest

def predict_sentiment(review, vocab, tokenizer, model):

        # clean
        tokens = clean_doc(review)
        # filter by vocab
        tokens = [w for w in tokens if w in vocab]
        # convert to line
        line = ' '.join(tokens)
        # encode
        encoded = tokenizer.texts_to_matrix([line], mode='freq')
        # prediction
        yhat = model.predict(encoded, verbose=0)
        return round(yhat[0,0])
Теги:
machine-learning
deep-learning
scikit-learn

1 ответ

1

Если вы выполняете тренировочный процесс в evaluate_mode(), модель является локальной переменной и не может использоваться совместно с параметром predict_sentiment(). Вы должны сделать model predict_sentiment() evaluate_mode() return, и пусть predict_sentiment() принимает это как четвертый аргумент.

Ещё вопросы

Сообщество Overcoder
Наверх
Меню