代码之家 › 专栏 › 技术社区 › Viet Phan

Tensorflow:分层Softmax实现

softmax hierarchical pytorch tensorflow python

Viet Phan · 技术社区 · 7 年前

我现在有用向量表示的文本输入,我想对它们的类别进行分类。因为它们是多层次的类别,所以我打算使用 Hierarchical Softmax .

Example:

 - Computer Science
     - Machine Learning
     - NLP
 - Economics
 - Maths
     - Algebra
     - Geometry

我不知道如何在Tensorflow中实现它。我遇到的所有例子都是使用其他框架。

谢谢

2 回复 | 直到 7 年前

greeness user1775765 7 年前

实际上,如果类别总数在数百到数千(少于50K)之间,则无需考虑使用分层softmax,该软件旨在更快地运行训练,以将分类为数百万个类别(例如,词汇表中的单词数)。

根据我的经验(使用朴素贝叶斯和神经网络),在训练时使用层次结构并不一定能提高分类质量。

Viet Phan 7 年前

最后,我改用Pytorch。它比Tensorflow更简单、更直接。

对于任何对实现HS感兴趣的人,可以查看我的示例说明: https://gist.github.com/paduvi/588bc95c13e73c1e5110d4308e6291ab
对于仍然想要Tensorflow实现的任何人来说,这是为您准备的: https://github.com/tansey/sdp/blob/87e701c9b0ff3eacab29713cb2c9e7181d5c26aa/tfsdp/models.py#L205 . 但这有点混乱,作者建议使用Pytorch或其他动态图形框架

推荐文章

Google User · Django管理员在`list_display中未显示`creation_date`字段`

4 月前

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

4 月前

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

4 月前

user29715306 · from_users=和chats=电视节目中的差异

4 月前

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

4 月前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

4 月前

prayner · 更新嵌套字典包含列表中的项

5 月前

Bringo Jr · 我可以在O(n)中解决这个问题吗?

5 月前

Dave · 如何在for循环中修改列表值

5 月前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

5 月前