seq2class是为使用lstm进行文本分类而开发的

seq2class的Python项目详细描述


Seq2Class是文本分类的一站式解决方案。使用序列模型(lstm)通过seq2class包简化了文本分类。这个包是通过利用lstm(长-短期内存)模型的功能构建的。

要求

Seq2Class支持Python3.6或更新版本。

安装

pip install seq2class

示例

这个软件包是为使用顺序模型进行文本分类而开发的。

data = 'movies.csv'
labels = 'title'
text = 'genres'

s = Sequence2class()
X_train, X_test, y_train, y_test = s.train_test_split(data, labels, text)
trained_model = s.fit_train(X_train, y_train, 500, 50, 7789, 5, 4)
prediction = s.predict(trained_model, X_test, y_train, 4)

其中movies.csv是包含文本和标签的培训文件。

参考文献

数据集信息


[1]  movies.csv dataset have been used for research purpose from this `*link* <http://files.grouplens.org/datasets/movielens/ml-latest-small.zip>`.

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java如何忽略缓冲读取器中在“”之后的行的其余部分,或行上的特定字符?   java在db中创建空对象或稍后保存   java如何实现UI无关的后台工作任务   java未能在Android中从BaseAdapter扩展的类中启动Tactivity?   java斐波那契迭代移动数组[]   安卓从文件读取提供了java。木卫一。StreamCorruptedException:无效的流标头:73720027   java计算矩形中的六边形数?   仅使用Java 1.5(或更早版本)读写XML   java如果所有元素都以相同的bucked结尾,为什么要进行大小调整?   java Apache POI Excel在xx中发现无法读取的内容。xlsx   swing我可以在普通java应用程序中使用GWTGUI吗?   来自自定义Java客户端的http删除请求的行为与邮递员不同   运行批处理文件时,java当前目录无效   使用TypeReference将java字符串转换为ArrayList<STRING>   documentlistener突出显示所有匹配词Java