tf оценщик input_fn повторно использовать набор данных - PullRequest
Новая
       5

tf оценщик input_fn повторно использовать набор данных

0 голосов
/ 22 ноября 2018

Я хочу разделить набор данных тензорного потока на поезд и набор проверок, чтобы использовать его в оценщике.Для этого мне нужно заранее создать набор данных, а затем передать его моему input_fn.

dataset = dataset.shuffle(10000)
train_dataset = dataset.take(train_lines).repeat()
valid_dataset = dataset.skip(train_lines).repeat()

def train_input_fn():
  return train_dataset

Однако это приводит к ошибке, поскольку набор данных взят из неправильного графика: 

Tensor("IteratorV2:0", shape=(), dtype=resource) must be from the same graph as Tensor("MapDataset_1:0", shape=(), dtype=variant).

Очевидно, что в Tensorflow все должно быть на одном графике.Как я могу повторно использовать набор данных для сохранения моего случайного разбиения?

...