Tensorflow 我如何训练我的数据集，有100000个元素和2个特性

我试图使用神经网络对amazon review数据集进行评论分类，但每次我尝试在google colab上运行它时，都会显示OOM错误，我理解这是由于内存溢出造成的。但是谁能告诉我应该如何训练我的数据集呢

---

附言-即使是小批量也不起作用

我们讨论的是数据集中的功能数量还是元素/项的数量？这些是要在数据集中训练的*元素。有两个不同的功能。我想帮助你，但我需要一些代码来做到这一点；如果我看不到代码，我就帮不了你。