leetcode grad cam text颜色分类

颜色分类leetcode用于文本的Grad-CAM，这是文本分类模型的实现。使用的模型是用于文本分类的1D-CNN。数据集经过精炼处理，重新标记以进行二元分类。输入特征为word2vec的精简版。需要Python>=3，tensorflow>=1.4,<2。

安装依赖：

pip3 install -r requirements.txt

在训练之前，需要准备word2vec二进制文件。可通过word2vec.sh脚本下载，将文件保存在word2vec/目录下：

./word2vec.sh

所有word2vec二进制文件需位于word2vec/目录中，例如：word2vec/GoogleNews-vectors-negative300-SLIM.bin。

训练

usage: train.py [-h] [--epoch EPOCH] [--batch-size BATCH_SIZE] [--lea...