Моя цель - экспортировать сохраненную модель тензорного потока, используя 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]
, которая, я подозреваю, связана с размером пакета, но я не уверен.
Обычно я бы расширил размеры входных данных, чтобы учесть размер пакета, но я не знаю, как это сделать в этой ситуации.