Ошибка тензора: список тензоров, когда ожидается единый тензор - PullRequest
Новая
       5

Ошибка тензора: список тензоров, когда ожидается единый тензор

0 голосов
/ 14 января 2020

Мой первый код с TF Estimator. Но когда я обучаю его, я получаю следующую ошибку:

TypeError: Список тензоров, когда ожидается одиночный тензор

Я использую функцию ввода следующим образом: 

def make_input_fn(df_x,df_y,num_epochs):
    #return tf.estimator.inputs.pandas_input_fn(
    return tf.compat.v1.estimator.inputs.pandas_input_fn(
    x=df_x,
    y=df_y,
    batch_size=50,
    num_epochs=num_epochs,
    shuffle=True,
    num_threads=1
    )

и затем я тренирую модель: 

model.train(input_fn=make_input_fn(train_x,train_y,10))
...