你好，想做单纯的分类。遇到问题。

[qizhang0302]

你好，我想做句子分类。如：上半年证金公司***** （句子），股票（label），单句对应一个分类标签。训练过程已经完成。但在eval过程时，遇到了。tensorflow.python.framework.errors_impl.InvalidArgumentError:assertion failed predictions must be in [0,1] [condition x <=y did not hold element-wise:x (ArgMax:0)= ] [5 5 5...] [y (auc/Cast_1:0) = ] [1] 请问还需要改什么代码呢

[qizhang0302]

如果我只想输出label，应该做哪些改动

[yaleimeng]

我在博览众码的基础上，终于解决了工业场景实用化的问题。使得模型预测的结果更直观、可追溯。让使用Bert做文本分类不再困难。 现已开源了代码库TextClassify_with_BERT，还在持续完善中。欢迎大家跟我沟通交流~