mayabot / mynlp

一个生产级、高性能、模块化、可扩展的中文NLP工具包。(中文分词、平均感知机、fastText、拼音、新词发现、分词纠错、BM25、人名识别、命名实体、自定义词典)
https://mynlp.mayabot.com/
Apache License 2.0
675 stars 90 forks source link

fasttext修改loss #41

Open ZiqiSun opened 2 years ago

ZiqiSun commented 2 years ago

在调用fasttext的时候,由于数据的不均衡,想要对loss进行改写,调用focal loss。请问有开发这个接口吗?

jimichan commented 2 years ago

loss对象创建是由createLoss更具参数创建的,你可以修改它,加入新的loss