给定已选择的标签,预测每个标签的概率

数据挖掘 算法 预言 监督学习 tfidf
2022-02-13 20:45:57

我有一组标签(〜10'000,将随着时间的推移而扩展)呈现给用户。在他选择了 3 个或更多标签之后,我想为每个剩余的标签预测用户也会选择这个标签的机会。我严格需要所有剩余标签的预测。预测应该很快,但准确性并不那么重要。训练数据将是用户已经选择了标签子集的情况。这里有什么好的方法?

1个回答

欢迎来到机器学习和数据科学。这是 RNN 很有用的经典情况。您可以自己从 Keras 训练一个,这将帮助您学习并获得更好的模型,但可能需要更多时间。或者,如果这是您的第一个项目,可能是 TextGenRNN 之类的东西(仍需要一些修改),您可以使用更预制的解决方案。

RNN理论

python中的示例

更多示例

TextgenRNN 库