上传者: 42106299
|
上传时间: 2022-03-28 22:52:25
|
文件大小: 8.52MB
|
文件类型: -
华侨城
OCTIS(优化和比较主题模型很简单)旨在训练,分析和比较主题模型,其最佳超参数通过贝叶斯优化方法进行估算。
安装
您可以使用以下命令安装OCTIS:
pip install -e .
您可以在requirements.txt文件中找到需求。
特征
我们提供了一组最新的预处理文本数据集(或者您可以预处理自己的数据集)
我们提供了一组著名的主题模型(经典模型和神经模型),或者您可以集成自己的模型
您可以使用几种最新的评估指标来评估模型
您可以使用贝叶斯优化相对于给定指标优化模型的超参数
我们提供了一个简单的网络信息中心,用于启动和控制优化实验
获取预处理的数据集
要获取数据集,您可以使用内置源之一。
from octis . dataset . dataset import Dataset
dataset = Dataset ()
dataset . load ( "oc