Tensorflow Estimator Train отправить пусто в функциях и ярлыках - PullRequest
Новая
       6

Tensorflow Estimator Train отправить пусто в функциях и ярлыках

0 голосов
/ 04 сентября 2018

Я пытаюсь передать X и Y на оценках Tensorflow, но

Получение размеров на features и labels как TensorShape([Dimension(None), Dimension(65536)]) и TensorShape([Dimension(None)]) соответственно. 

    def model_fn(features, labels, mode):

    #labels tensor is of None Dimension shape.

    model = tf.estimator.Estimator(model_fn)

    # Define the input function for training
    input_fn = tf.estimator.inputs.numpy_input_fn(
    x={'images': X_train}, y=np.ones(45,np.float),
    batch_size=batch_size, num_epochs=20, shuffle=True)
    # Train the Model
    model.train(input_fn, steps=num_steps)

Размерность для X_train равна (45,65536), а у просто как единицы. 

num_steps = 22
batch_size = 5
...