如何解决“仅在启用急切执行后才支持RuntimeError：dataset .__ iter __（）。”

Question

我想批处理输入数据，但是会发生错误。如何调整代码？

def input_fn(data,batch_size): 

    Continuous_cols = {k: tf.constant(data[k].values) for k in Columns}

    label = tf.constant(data[Label].values)

    dataset = 
    tf.data.Dataset.from_tensor_slices((dict(Continuous_cols.items()), label)) 
    dataset = 
    dataset.batch(batch_size)

    return dataset

Answer 1

尝试在Google Colab中将您的运行时类型更改为TPU 。