AttributeError: у объекта 'list' нет атрибута 'argmax', и numpy .AxisError: ось 2 выходит за границы массива измерения 1 - PullRequest
Новая
       4

AttributeError: у объекта 'list' нет атрибута 'argmax', и numpy .AxisError: ось 2 выходит за границы массива измерения 1

0 голосов
/ 08 марта 2020

У меня есть функция для прогнозирования модели NMT keras, и она работает нормально, но она читает файл, а затем сохраняет прогноз в другом файле

, но я хочу, чтобы пользовательский ввод и сделал прогноз

полный код 

from keras.preprocessing.sequence import pad_sequences
from keras.models import load_model
import sys
import warnings
import argparse
from seq2seq_utils import *

ap = argparse.ArgumentParser()
ap.add_argument('-max_len', type=int, default=95)
ap.add_argument('-vocab_size', type=int, default=31500)

args = vars(ap.parse_args())
MAX_LEN = args['max_len']
VOCAB_SIZE = args['vocab_size']


def load_test_data(user_input, X_word_to_ix, max_len):
    user_input = input()
    X = [text_to_word_sequence(x)[::-1] for x in user_input.split('\n') if 0 < len(x) <= max_len]
    for i, sentence in enumerate(X):
        for j, word in enumerate(sentence):

            if word in X_word_to_ix:
                X[i][j] = X_word_to_ix[word]
            elif word in X_word_to_ix is None:
                X[i][j] = None
            else:
                X[i][j] = X_word_to_ix['UNK']

    return X


model = load_model('model.h5')
model.get_weights()
X, X_vocab_len, X_word_to_ix, X_ix_to_word, y, y_vocab_len, y_word_to_ix, y_ix_to_word = load_data('english.txt',
                                                                                                   'french.txt',
                                                                                                   MAX_LEN,
                                                                                                   VOCAB_SIZE)

saved_weights = find_checkpoint_file('.')
print('please enter the value')
user_input1 = input()
if len(saved_weights) == 0:
    print("The network hasn't been trained! Program will exit...")
    sys.exit()
else:
    X_test = load_test_data(user_input1, X_word_to_ix, MAX_LEN)
    print(type(X_test))
    X_test = pad_sequences(X_test, maxlen=4, dtype='int32')
    print(type(X_test))
    arr=np.array(X_test)

    model.load_weights(saved_weights)
    predictions = np.argmax(model.predict(arr))
    # predictions = np.argmax(list(model.predict(X_test)))
    print(type(predictions))

    # predictions = np.argmax(model.predict(X_test),axis=0)
    sequences = []
    print('1')
    for prediction in predictions:
        print('2')
        sequence = ' '.join([y_ix_to_word[index] for index in prediction if index > 0])
        print(sequence)
        l = sequences.append(sequence)
        print(l)
    np.savetxt('test_result.txt', sequences, fmt='%s')

AttributeError: у объекта списка нет атрибута argmax

Во время обработки вышеупомянутого исключения , произошло другое исключение:

numpy .AxisError: ось 2 выходит за границы для массива измерения 1

почему? и спасибо за помощь

...