【发布时间】:2017-09-05 02:09:45
【问题描述】:
示例:我在职位描述中有一句话:“英国Java高级工程师”。
我想使用深度学习模型将其预测为 2 个类别:English 和 IT jobs。如果我使用传统的分类模型,它只能在最后一层使用softmax 函数预测 1 个标签。因此,我可以使用 2 个模型神经网络来预测两个类别的“是”/“否”,但如果我们有更多类别,那就太昂贵了。那么我们是否有任何深度学习或机器学习模型可以同时预测 2 个或更多类别?
“编辑”:传统方法使用 3 个标签,它将由 [1,0,0] 编码,但在我的情况下,它将由 [1,1,0] 或 [1,1,1] 编码]
示例:如果我们有 3 个标签,一个句子可能适合所有这些标签。因此,如果 softmax 函数的输出为 [0.45 , 0.35 , 0.2 ] 我们应该将其分为 3 个标签或 2 个标签,或者可能是一个?
我们这样做的主要问题是:分类为 1、2 或 3 个标签的良好阈值是多少?
【问题讨论】:
-
使用两个模型?
-
如我所说,如果我们有 10 - 20 个类别,那就太贵了
标签: machine-learning tensorflow nlp deep-learning