Как я могу исправить свой код классификатора оценщика - PullRequest
Новая
       1

Как я могу исправить свой код классификатора оценщика

2 голосов
/ 30 марта 2019

Привет, я новичок в Tensorflow и я практикуюсь с библиотекой tenSenflow.estimator.В основном я запустил встроенный алгоритм tf.estimator.DNNClassifier ниже 

import tensorflow as tf

def train_input_fn(features, labels, batch_size):
    """An input function for training"""
    # Convert the inputs to a Dataset.
    dataset = tf.data.Dataset.from_tensor_slices((dict(features), labels))

    # Shuffle, repeat, and batch the examples.
    return dataset.shuffle(1000).repeat().batch(batch_size)

# Feature columns describe how to use the input.
my_feature_columns = []
for key in landmark_features.keys():
    my_feature_columns.append(tf.feature_column.numeric_column(key=key))


# Build a DNN with 2 hidden layers and 10 nodes in each hidden layer.

classifier = tf.estimator.DNNClassifier(feature_columns=my_feature_columns, hidden_units=[10, 10],n_classes=10)


dataset = train_input_fn(landmark_features, emotion_labels, batch_size = 1375 )

Однако я продолжаю получать следующую ошибку: 

INFO:tensorflow:Using default config.
WARNING:tensorflow:Using temporary folder as model directory: /tmp/tmpc_tag0rc
INFO:tensorflow:Using config: {'_model_dir': '/tmp/tmpc_tag0rc', '_tf_random_seed': None, '_save_summary_steps': 100, '_save_checkpoints_steps': None, '_save_checkpoints_secs': 600, '_session_config': allow_soft_placement: true
graph_options {
  rewrite_options {
    meta_optimizer_iterations: ONE
  }
}

Есть какие-либо идеи о том, что я могу сделать, чтобы исправить свой код?

...