web-dev-qa-db-ja.com

Tensorflow:階層型Softmaxの実装

現在、ベクトルで表されたテキスト入力があり、それらのカテゴリを分類したいと考えています。それらはマルチレベルのカテゴリーなので、Hierarchical Softmax

Example:

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

Tensorflowで実装する方法がわかりません。私が出会った例はすべて、他のフレームワークを使用しています。

ありがとう

14
Viet Phan

最後に、Pytorchを使用するように変更しました。 Tensorflowよりも簡単で簡単です。

4
Viet Phan

実際には、カテゴリの総数が数百から数千の範囲(50K未満)である場合、階層的ソフトマックスの使用を考慮する必要はありません。語彙の単語数)。

私の経験では(ナイーブベイジアンおよびニューラルネットワークを使用)、トレーニング時に階層構造を利用しても分類の品質が向上するわけではありません。

ただし、とにかくHierarchical Softmaxの実装に興味がある場合は、別の話です。

12
greeness