如何修复“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。