search

Tencent / NeuralNLP-NeuralClassifier

An Open-source Neural Hierarchical Multi-label Text Classification Toolkit
Other
1.83k stars 402 forks source link

训练hierarchical classifier with HMCN应该如何修改分类的层级 #122

Open Walker-DJ1 opened 8 months ago

Walker-DJ1 commented 8 months ago

原始文件中配置如下："HMCN": {"hierarchical_depth": [0, 384, 384, 384, 384],"global2local": [0, 4, 55, 43, 1]}，假设用户的分层只有两层，应该如何配置分层深度的大小和全局到局部的尺寸？

MagiaSN commented 1 month ago

hierarchical_depth是模型超参数，列表长度=标签层数+1，第一个元素固定为0，其他可以自行设置；global2local第一个元素固定为0，其他元素是每一层标签的数量