Как создать конструктивные столбцы для модели RNN в Tensorflow? - PullRequest
Новая
       6

Как создать конструктивные столбцы для модели RNN в Tensorflow?

0 голосов
/ 06 апреля 2019

Моя цель - экспортировать сохраненную модель тензорного потока, используя tf.estimator.export_saved_model.

У меня есть две функции, назовем их feature_1 и feature_2. Каждый из них имеет форму (5, 181). Они являются последовательными элементами, поэтому 5 = временные шаги и 181 = вектор из 181 числа с плавающей запятой. 

feature_columns = []
keys = ['feature_1', 'feature_2']
for key in keys:
    col = tf.contrib.feature_column.sequence_numeric_column(key,shape=(5, 181))
   feature_columns.append(col)


features = tf.feature_column.make_parse_example_spec(feature_columns)
recvfn = tf.estimator.export.build_parsing_serving_input_receiver_fn(features)
estimator.export_saved_model(path_to_saved_model, recvfn)

Я получаю эту ошибку ValueError: Input 0 of layer sequential is incompatible with the layer: expected ndim=3, found ndim=2. Full shape received: [None, None], которая, я подозреваю, связана с размером пакета, но я не уверен.

Обычно я бы расширил размеры входных данных, чтобы учесть размер пакета, но я не знаю, как это сделать в этой ситуации.

...