基于Python+JavaScript的面向文本分析的交互式主题建模可视化分析系统- C. |& b( {- P* x P& A* l. M _
" [; J2 L8 K8 `; p+ o面向文本分析的交互式主题建模 ( L+ F- W! Q' F) o 7 D5 s$ G/ X/ L* S目录 6 C( U/ k I2 n面向文本分析的交互式主题建模 1* X$ N5 E ?, K! O4 u _
一、绪论 26 U. h: o! g$ `6 T
1.1 本课题的研究背景和意义 2 ; H5 _ H1 b. p0 i1.1.1 主题模型的发展及研究现状 2# O! c$ c# @! I$ g7 m
1.1.2 目前存在的问题 3 + m; `8 t! Z" b, x1.1.3 本课题的研究意义 3 7 J4 G+ H5 B% e" Q6 D1.2 研究内容和主要工作 30 Z. C' U' S3 D+ L
1.3 本文的组织结构 3 , a5 ~% j. Z" |0 \% n二、核心算法 3% f6 N7 n5 W$ [+ w
2.1 文本预处理 4 " v6 b; I2 S/ z4 Z6 ~2.2 大型语料库的内存优化 6 Z% ~' Y! g% E% `- Y
2.3 UMAP 数据降维&可视化 7/ M9 p+ j4 s5 E/ `% n" k
三、系统设计与实现 9' O) I+ M0 J4 F# D/ K" q( P# b# A$ y
3.1 系统介绍及流程图 9' d n8 j0 t1 X4 E- v; x% j
3.2 后端实现过程 10 & X3 G7 v( m$ |0 u3.2.1 框架介绍 103 f1 y. Z- _0 z' r
3.2.2 数据库 ORM 映射类 10 d- @# W6 j% B
3.2.3 Pydantic 模型类 12 1 K r l/ n0 |) |& X, h. p, Q; e3.2.4 TextPreprocessing 文本预处理类 129 Q' t* l7 h: k" d+ B& ^$ N
3.2.5 SSNMFTopicModel 非负矩阵分解类 12 0 W, t# g5 ]) D' W, }8 ?; H3.2.6 TopicModelTrainingTask 训练任务类 12 V$ b' j) o5 K: @3.2.7 Web API 接口 13: H' t. Z3 `( r8 f0 t' ]* f
3.3 前端实现过程 16, G, F' m" U# S
3.3.1 框架介绍 169 t% N5 E6 ]* v6 d
3.3.2 介绍界面 167 b8 y9 W5 L# ^4 J8 {% k; j2 b3 r
3.3.3 Bilibili 视频评论爬虫界面 176 E T9 ]& I$ r" R9 l
3.3.4 语料库查看/选择界面 18 6 ^9 a' j+ K! E# Z4 R! C3 `* \8 m3.3.5 侧边栏主控制界面 19 / M: `& ^* @7 a; M# ^3.3.6 右侧训练状态 Tab 19 ( F6 n7 s2 V: q8 b* B8 Z$ A3.3.7 词云图 Tab 196 t* C4 _* i! r6 ~: F0 S, G1 }& w) W
3.3.8 NMF 迭代误差折线图 Tab 206 m4 F/ ?7 P+ d2 ^7 }
3.3.9 主题聚类可视化 Tab 202 \3 L( L1 g) H; d
3.3.10 文档详细信息抽屉 22: R' j/ l, B% D7 v2 H
3.3.11 主题详细信息抽屉 23 # _( g7 f% l! A4 Z# \" n1 A3.3.12 用户交互 239 L) y6 m4 _8 i6 F4 o5 ?4 ]8 o
3.3.13 新文档主题分布预测 29 : q( ]6 S8 r, S- F1 ?1 {8 K四、算法定量分析及比较 298 c0 R2 J8 e% ^3 P
4.1 性能及收敛速度 30 , Z. r6 |* q `7 q- [! {4.2 多次运行的一致性 30 4 T& x# ~" C2 h: k2 b( J% S9 U五、使用案例 31 ; p' I# p. U# j5 ~6 \5.1 Bilibili 视频评论数据 31 4 |) I/ A+ `/ s( @4 s* _5.2 外卖用户评价数据 39% T& M' ^2 ^ _% r- O+ _
5.3 新冠病毒新闻数据 441 c+ X3 d1 Y* s+ S
六、总结及展望 47 ; C, Y7 ]; ^* Z: v# j一、绪论5 E/ k0 r/ z" P6 @
1.1 本课题的研究背景和意义( `8 u B% P( [! Y3 s9 Q/ k) h
近年来,随着网络以及社交媒体的不断发展及活跃,互联网上每天都会产生大量的文本信息。在这些海量的文本数据中,对其进行分析并提取出有价值的内容,并由此指导用户的决策过程一直是研究的热点领域。在解决这些问题的各种方法中,主题建模方法(从文档语料库中发现语义上具有意义的主题)在数据挖掘/机器学习和可视化分析领域得到了广泛的应用。顾名思义,这是一个自动识别文本对象中存在的主题并派生文本语料库显示的隐藏模式的过程。 , o* b# U$ O7 I5 D. C2 o主题建模不同于使用正则表达式或基于字典的关键字搜索技术的基于规则的文本挖掘方法。这是一种无监督的方法,用于查找和观察大型文本集中的一堆单词(称为“主题”)。主题可以定义为“语料库中共同出现的术语的重复模式”。 9 _) @. o. m8 V- o' ]主题模型有着广泛的应用场景,特别是对于文档聚类,组织大量文本数据,从非结构化文本中检索信息以及选择功能非常有用。例如,《纽约时报》正在使用主题模型来提升其用户–文章推荐引擎;各种专业人士正在针对招聘行业使用主题模型,他们旨在提取职位描述的潜在特征并将其映射到合适的候选人。" s( F8 I; ^( p
在本节接下来的内容中,我们将回顾主题模型的发展及研究现状,提出目前存在的问题,以及本课题的研究意义。; e/ m; q9 h/ D
1.1.1 主题模型的发展及研究现状 ; M% P& q1 ]( }* b; Z早在上世纪 80 年代,研究者们常常使用 TF-IDF 方法来进行文档的检索和信息的提取。在此方法中,先选择一个基于词或者词组的基本词汇,然后对于语料库中的每个文档,分别统计其中每个单词的出现次数。经过合适的正则化后,再将词的频率计数和反向文档频率计数相比较,得到一个词数乘以文档数的向量 X。向量的每个行中都包含着语料库中每个文档的 TF-IDF 值。 1 R! ~5 Z7 f6 i: r1 d8 x9 k* C但是整个 TF-IDF 算法是建立在一个假设之上的:一个单词出现的文本频数越小,它区别不同类别文本的能力就越大。这个假设很多时候是不正确的,尤其是在引入 IDF 的过程中,单纯地认为文本频率小的单词就越重要,文本频率大的单词就越无用,显然这并不是完全正确的。其也不能有效地反映单词的重要程度和特征词的分布情况,因此精度有限。 " S5 x* B+ K* T9 @" [) T: l$ }8 h上世纪 90 年代,Deerwester S 等人提出了潜在语义分析(LSA),它是一种用于知识获取和展示的计算理论和方法,出发点就是文本中的词与词之间存在某种联系,即存在某种潜在的语义结构。Thomas Hofmann 于 1999 年提出了概率潜在语义分析(pLSA),并在文中描述了 pLSA 与 LSA 的区别,即 LSA 主要基于奇异值分解(SVD),而 pLSA 则依赖混合分解。他随后进行了一系列实证研究,并讨论了 pLSA 在自动文档索引中的应用。他的实证结果表明 pLSA 相对于 LSA 的表现有明显进步。* z/ ?/ S( h, |7 g, X
年,潜在语义分析被 Jerome Bellegarda 提出使用在自然语言处理中。2003 年 Andrew Y. Ng 等人在论文中提出用于 pLSA 的 aspect model 具有严重的过度拟合问题,他们提出了隐含狄利克雷分布(LDA),这可以看作是结合了贝叶斯思想的 pLSA。LDA 是目前使用的最常见的主题模型。- z' b9 A6 t" ^; ~
年,针对当前主题模型可以为文档构建可解释的向量表示,而词向量在句法规则方面十分有效,Christopher E Moody 提出了lda2vec,一个学习密集单词向量的模型,将上述两类模型的优点结合。他们的方法很容易融入现有的自动微分框架,并允许科学家使用无监督文档表示,同时学习单词向量和它们之间的线性关系。0 y$ Q4 P( W% \! i3 e+ s* ^
1.1.2 目前存在的问题 1 x- ?/ W& O7 G8 B主题模型是一种无监督学习模型,其结果的好坏取决于所选的模型参数和训练集,并且具有很高的不确定性。用户通常无法在训练过程中对模型的结果进行修正,特别是一些专业领域,用户无法向主题模型提供一些领域知识来提高主题建模的质量。已有一些研究提出,通过对主题模型添加约束的形式来解决这一问题。Hu 等人提出了交互式主题模型,然而主题建模的结果十分不直观,从中寻找不恰当的结果并添加合适的约束非常耗时、费力。同时现在基于概率图模型的主题建模算法结果往往具有不确定性,很难引入用户的交互反馈操作,比如让用户交互调整建模的中间结果和参数,从而得到更优的主题分析结果。 |9 Y o. Z( R3 @6 `8 u1.1.3 本课题的研究意义 4 ~/ n1 T" Q3 l! G% C( L本课题的研究意义是对当前主题建模算法进行优化改进,解决算法结果的不确定性和用户交互反馈引入的困难性这 2 个问题,将可视化分析技术与主题模型相结合,提供有效的交互手段,让人们充分参与到分析主题模型的结果中来,利用人的认知能力,从数据中挖掘有效信息,达到基于用户驱动的文本主题模型交互优化。 ' B/ [1 n- G- R9 g1.2 研究内容和主要工作3 ?# l6 z" D% ^$ e; i) a
本课题的研究主要内容是在国内外交互式主题建模的研究基础上,设计一个面向文本分析的交互式主题建模可视化分析系统。本文转载自http://www.biyezuopin.vip/onews.asp?id=15157系统使用非负矩阵分解(NMF)来作为主题建模的主要算法,并参考了 Jaegul Choo 等人提出的 UTOPIAN 主题建模可视化分析系统,对其中的半监督非负矩阵分解(SS-NMF)算法和多种用户交互方式进行了实现及改进。& H) J5 l7 C9 ~% ]* r
本系统采用 Web 技术,后端算法和接口使用 Python 语言和 FastAPI Web 框架编写,前端界面使用 React&Ant Design 框架编写。主要工作可分为以下几个部分:1 `3 _& D: \: P$ Q& A. M) G
语料库的爬取、收集和整理,并写入数据库。 ! Y$ s- R; Y( j0 [( s& i后端主题建模相关算法的编写。 J( |& Z' t$ f% s
后端 Web API 接口的编写。 z/ @& Z! E& G- X0 C7 [
前端界面的设计。4 i3 f) g1 q- X( ^
前端与后端的对接,调用后端数据接口,并将数据可视化。 5 m) e+ X6 d: w: N- M! d前端用户交互逻辑的编写。 8 q; S. F& P: G+ c0 Y$ M6 V4 @/ E: N$ f; V; g3 y+ E9 \
from typing import List ! c3 s3 R# e3 ]8 e; \% K, l% G$ _- U B9 J* E
from fastapi import Depends, FastAPI, BackgroundTasks, Response, Cookie, HTTPException . s* r3 g. h# `6 C9 ?9 t9 W2 o I& Dfrom fastapi.middleware.cors import CORSMiddleware {6 l; V9 U$ J m, pfrom sqlalchemy.orm import Session W1 x. T1 M. _* R2 N
from typing import Optional, Dict : e7 \2 E! e, F0 pimport crud, models, schemas / J- i5 p: H% x7 L% q* X; ffrom database import SessionLocal9 J$ L. U6 r: F& P' ?8 |) }1 |
from TopicModelTrainingTask import * . q7 R1 A2 W4 j, p9 u" q, Rimport uuid) ]( |1 c/ R* z, q& J
r$ e) F M" ?5 S/ Capp = FastAPI() . w* U/ h( \ Z2 O9 n# S8 [% o& y. E2 J8 k7 n2 T5 T9 R/ ~
# 配置跨域问题 * A: r! q; h5 K* K8 @/ n& jorigins = [$ i0 b& ], _/ A/ J$ ?
"http://localhost:8000",! x1 i" c2 r$ Q6 F
"http://localhost:3000" " f) h" V8 B" v* _, W% e6 x6 l3 @]- A ~: T1 s. i' m3 ]
% F; t3 @% [2 a- W& W6 ^- w4 S; o: z
# 全局主题模型训练对象; | V$ |% @+ e* _' b/ D! D6 i
topic_model_training_tasks: Dict[str, TopicModelTrainingTask] = {} " P7 y* e6 K3 H0 ?/ ?" Q; F, V# P $ N9 }2 M7 N9 ~9 q3 q9 kapp.add_middleware( 8 @$ x1 r! c) j+ S) J- R CORSMiddleware, ! u7 X# @% K0 z$ m; e E allow_origins=origins, ; K/ j2 r" n" \ Z allow_credentials=True,: F8 u+ `- O( q" ^" i) l* c/ F
allow_methods=["*"],8 K$ \. A% b0 y
allow_headers=["*"],9 M4 i' H' g7 X
) ! K) `6 q" j6 G6 b4 `, t! {* ]' g' a g
+ _# k; d: j2 k' v: P! ]3 w/ J# 数据库连接依赖+ d; x7 v6 I0 x- I/ t* a e' ^
def get_db():6 D( l& \* B% b) n0 I; T
try: - v& ~7 O. f' A, e3 n& u2 Z% k$ k* d$ M db = SessionLocal()7 {* I/ e8 x" o
yield db5 _1 U" w( } r! i4 U# c& o
finally: - t: X. ^5 q& R( \7 ?" _+ @* g db.close(), @5 @0 C6 P- e
- b2 j5 j' x1 o! \9 `
; {) ~& h5 h/ w' p4 m! l# h1 w: Q@app.get( ) z, _' \; [$ e- O9 F9 g "/bilibilivideos/", " d/ ~4 G4 S9 L9 a response_model=List[schemas.BilibiliVideo], 4 ? p' p" Y+ t0 z |9 s tags=['bilibili视频评论语料库数据接口']5 T7 C! S! ]- E
)! e2 w2 q' u% b d. ?% Z
def get_bilibili_videos(db: Session = Depends(get_db)): 6 o0 p7 ?6 G# o, X db_videos = crud.get_bilibili_videos(db) 9 }( O' c, Y/ B; G# V return db_videos: h2 ?* c, R2 }* A9 y# S
' R. r7 h2 g _: s9 t7 s) q$ ^
2 H8 s( h. a1 |) L- R@app.get( 4 U8 z! ^9 G4 W6 k "/bilibilivideos/{vid}/", |7 b+ R4 C; T, `: b$ z
response_model=schemas.BilibiliVideo, . V/ n* X' z0 Z tags=['bilibili视频评论语料库数据接口']( ]0 N' @$ m+ e' y, z. x; n
) 1 `0 z: c0 \. q) @def get_bilibili_videos_by_vid(vid: int, db: Session = Depends(get_db)):' s0 ?1 A* I1 e1 R0 {
db_video = crud.get_bilibili_videos_by_vid(vid, db)1 k* A$ h% ~* x9 ]6 B! N4 W
return db_video, F6 I0 Y, m9 U I5 _1 Y
/ |: o6 ]1 p: ~( U' N9 e% t( P1 z+ k: g
@app.get(% q- U* i- j6 _1 k
"/bilibilivideos/{vid}/comments/",) p6 k2 ?* H' e8 v9 c; d. v5 @! u
response_model=List[schemas.BilibiliVideoComment], 8 R. z' p5 [3 R6 a+ k tags=['bilibili视频评论语料库数据接口'] / e$ e* D O# o+ u( r, M J- B' T2 l) 6 C% l# \( `# s0 W+ Xdef get_bilibili_video_comments_by_vid(vid: int, db: Session = Depends(get_db)): # ?7 d4 r" H/ l- Q. C r; q3 { db_comments = crud.get_bilibili_video_comments_by_vid(vid, db) ' q9 Z( t6 J# P1 m" Z' Q return db_comments N- O% o0 @" q+ q, |
R3 e. P! z( h9 t+ N) Q8 \6 U; Z/ u
@app.get( ) N4 p V5 a9 m; \/ E "/onlineShoppingReviews/",0 f. _- q8 I$ d& @8 g: o; s
response_model=List[schemas.OnlineShoppingReview],. R6 `' A9 ?& m
tags=['电商购物评价语料库数据接口'] 8 v/ r4 Z/ e+ C9 x): A7 v, e8 P6 c1 h( E- n
def get_online_shopping_reviews(db: Session = Depends(get_db)):- G9 U; h' m3 U
db_reviews = crud.get_online_shopping_reviews(db) 7 O; ~* A7 O9 @& Q, C( L return db_reviews . R$ z; W. B7 t# K- r: s; \ ) B8 q5 ^9 \% g# q5 |) f o ) E, B' @* n2 ~8 G4 q2 A@app.get( 8 P) h" P, E! p' n% T "/takeawayReviews/",1 s4 W r& z$ H4 i) E
response_model=List[schemas.TakeawayReview],# t1 X! I( L8 a( c x6 q
tags=['外卖评价语料库数据接口'] ; ]0 U# T9 d. C)7 v; _- m# d. _1 i' Z. B
def get_takeaway_reviews(db: Session = Depends(get_db)): ; ]9 F( ]7 {$ U+ Y db_reviews = crud.get_takeaway_reviews(db). {' O, w7 s' I, _- c {- d
return db_reviews , A+ t1 L( }# H) ?5 q" B ) ]( C5 F, n+ y: U ) W6 Z7 k) i: {: s@app.get( ( c* ?2 D2 m/ k+ r. Z7 u "/chineseLyrics/",. e- Q& s% D" u; M+ N
response_model=List[schemas.ChineseLyrics],* x; ]/ x, o: M3 a7 n
tags=['中文歌歌词语料库数据接口']1 K" e- m' J2 P* C, ]/ O! i
)- v3 e# O& o5 }* n a( w% g) t! ]
def get_chinese_lyrics(db: Session = Depends(get_db)):; W9 y1 v; z* G5 X9 L' c* @
db_reviews = crud.get_chinese_lyrics(db) * w! |4 ]# v/ l: C7 G8 a& k" l return db_reviews. V8 ~, |2 {" @! r; O
$ X+ J! A/ d1 c* v e: q* @
+ E2 e( V: i! S8 D. Q1 i/ S
@app.get( 2 c7 l) K+ v% ]- a5 u "/COVID19News/", . w( ]% O% D1 v4 B response_model=List[schemas.COVID19News], " K* e8 W+ ]& V7 W a tags=['新冠病毒新闻语料库数据接口']- j" L4 N7 K9 w! q- N
) A$ G7 I o; C; ]# G( @$ M
def get_COVID19_news(db: Session = Depends(get_db)): / w# ] O; S7 d6 p% J9 ], a db_reviews = crud.get_COVID19_news(db) " }* d+ O; w* a# t) v* O return db_reviews D. c! g. S+ ~* `
- r7 h# a2 V% y/ R# C, I! ~- A* ]% j$ Y8 s6 O$ U
# 为每个用户设置uuid标识符,并存入Cookie, [. g: O+ ~- L1 k% ]
def create_user_uuid_and_set_cookie(response: Response):! f9 n' K- b. D
user_uuid = str(uuid.uuid1())+ v4 {0 `/ s$ E2 L: s7 b
response.set_cookie(key="user_uuid", value=user_uuid) i: i1 T) Q7 a# l& v; s! {8 q
return user_uuid 9 D6 R/ R4 O: A) I! I; x! |' C $ y7 P6 q% h& m" L: N" K1 @& z4 b0 h - Y/ T) ^5 W9 q. L0 M7 p! N- b@app.post(' h) p1 C' \8 _0 ]/ i
"/calculate/preprocessing/", : W$ f9 z* t O tags=["计算模块数据接口"]. |/ B& _& W8 u
)1 {0 k5 c: P* Z6 m, P* A5 p- c8 p
async def run_text_preprocessing_task(text_preprocessing_params: TextPreprocessingParams,/ f5 P, T9 X; m, Z1 Z
background_task: BackgroundTasks, 4 V# h: Y+ a) N) B0 k& | response: Response, : B9 ^, \7 _0 b! M" o6 } user_uuid: str = Cookie(None)):. s" [ H/ Y3 _: |7 H$ Q1 T
if not user_uuid:2 J* Z; E: a1 w9 k
user_uuid = create_user_uuid_and_set_cookie(response)7 ]1 O0 V1 }. i2 Z& d
% Y9 s" L P/ f0 ]/ t i5 ] B
global topic_model_training_tasks2 H. u* N( H e1 k g8 M& B
task = TopicModelTrainingTask(text_preprocessing_params)7 C+ w3 c1 o5 y! @
topic_model_training_tasks[user_uuid] = task " \+ Z! @) Y6 \7 Z0 ~9 E2 k4 d. c- |+ X1 R
background_task.add_task(task.preprocessing) 5 t. ] [* }3 m5 p return {"message": "已添加文本预处理后台任务"}* B! Y+ `$ Q+ T0 z) k
( k" X1 e" F1 B) K% w0 d4 ^: \' a3 S
@app.post(8 S1 u3 R3 @% t7 m/ R
"/calculate/nmftraining/", + m* Z1 N7 L" l' F C3 t: D9 A" L( v tags=["计算模块数据接口"]& y7 @+ ~/ r& X5 Y) `6 i, i A
) . ~9 M/ G& C" ^; B% z- ~9 Fasync def run_nmf_training_and_tsne_task(nmf_training_params: NMFTrainingParams,0 C' X/ z; C( w6 i
background_task: BackgroundTasks,$ E: |4 U* Q2 d- ^1 f3 A
response: Response, * C3 P i, \- c$ Y% y user_uuid: str = Cookie(None)): 6 `; B0 _4 x0 `0 l% \ if not user_uuid:6 A7 {* H) P8 w% r& n: N g, c$ j4 _
user_uuid = create_user_uuid_and_set_cookie(response): ~* a& E1 \6 m. `! b
$ q) C# a ]1 r" Y* W
global topic_model_training_tasks % k; \4 T6 J% Y& @. q" c6 A$ T" {+ u/ O
if user_uuid and user_uuid in topic_model_training_tasks:7 V2 H' U- H, u2 G, I3 ~; i7 v
task = topic_model_training_tasks[user_uuid] {8 H5 P6 o; ~( d3 A if task.text_preprocessing_progress.status_code == 2: t8 o/ |7 h; e7 w4 M background_task.add_task(task.nmf_training, nmf_training_params)( W0 e+ T) R' R1 b" M
return {"message": "已添加NMF主题模型训练后台任务"} . R3 c6 P( K1 I' x& c, g& | else: ! q$ Z/ f$ [) k n' g- P4 ` raise HTTPException(status_code=404, detail="请先进行文本预处理任务!")/ `0 d2 _! n0 Y4 U' t+ O: S
else: ( a$ @* ] E& F# L e raise HTTPException(status_code=404, detail="未找到相应的训练对象!")& e* x) O1 X- V& D3 Z+ v+ h w
2 l1 T$ p) e# ~4 a$ p6 i( P# d8 |7 Z m/ p5 S
@app.get( # @; V# v/ [0 C+ w1 P$ v "/calculate/nmftraining/keywordsearch/", ; U4 q4 x* Q$ `4 G- Y+ O$ ^ response_model=KeywordSearchResult,6 J& t* d. ~0 {* B
tags=["用户交互"] , {) X9 h! X5 |5 o) 8 w7 g9 U; E1 J; B% i4 Wasync def search_keyword(search_text: str, 0 e9 Y- i. o) @& {/ P; R response: Response,) }: g( D7 _. r7 I
user_uuid: str = Cookie(None)):' v& B- [6 q2 r$ T+ W. E7 d
if not user_uuid: ! O* u% M# z" H6 i1 ~ user_uuid = create_user_uuid_and_set_cookie(response) ( O+ ~5 P9 m0 Z" p. e& Y8 u' K0 X* B* ^* L
global topic_model_training_tasks 7 }3 f/ z E7 y [5 V0 Q& R/ b" \& w5 u3 R: P
if user_uuid and user_uuid in topic_model_training_tasks: + y, \8 l& ^5 D. M: J) w task = topic_model_training_tasks[user_uuid] , N8 U2 _+ u: `" g$ T0 h if task.text_preprocessing_progress.status_code == 2:2 A- i0 t1 @8 o; l& X
search_text = search_text.lower()4 m& `3 e7 M4 }1 ?* k# Q) J
bag_words = task.text_preprocessing.bagWords , `# n7 Z" l' X0 t8 L keyword_search_result = KeywordSearchResult() $ f9 H7 U4 }1 t for wi, word in enumerate(bag_words):) T1 q" w; j* B( e' v; {' u" z3 [
if search_text in word: " ^3 ~% K# P5 i* |% R# ] keyword_search_result.word_id_list.append(wi)4 Z5 m" C/ a9 w7 L. j# Z% r8 J
keyword_search_result.word_list.append(word) 5 G5 w# s M1 U3 |4 w( s return keyword_search_result " S! J6 o- m% H' y# x else: # _* }2 O9 B r8 U raise HTTPException(status_code=404, detail="请先进行文本预处理任务!"). e5 T& H$ W- a: h& m
else: , x, _; a, u& L0 l' H$ B" V raise HTTPException(status_code=404, detail="未找到相应的训练对象!") 2 j6 }$ o; R. n2 R ! ~8 p W, C, g7 P4 k$ b* ? # B7 S# c: q7 y@app.post( 1 O- d, V3 H. D$ `" C! z "/calculate/nmftraining/topickeywordoptimization/", # {8 c( a0 B7 i; A0 N tags=["用户交互"]# I+ ]7 O3 u! Q0 r( o+ X" {5 R
) + z4 s9 j& n! L- Basync def run_topic_keyword_optimization_task(tko_params: TopicKeywordOptimizationParams,, }% G3 l: `, ~
background_task: BackgroundTasks, : l6 l; r' _5 f response: Response,$ `% w& l. ?" V: F# U
user_uuid: str = Cookie(None)):1 K4 t v6 K# H/ f# Q
if not user_uuid:* D( Q- P7 F6 L: d4 T/ o
user_uuid = create_user_uuid_and_set_cookie(response)& S7 B4 n7 D4 |* S7 R) `3 a
& g8 {5 W5 g5 r; }1 S7 c) }, T global topic_model_training_tasks 1 h; p% u, D @8 I# m2 R7 ?1 r6 G$ y1 p- M: e; N6 t$ ]/ U# ~/ C
if user_uuid and user_uuid in topic_model_training_tasks:8 f3 v' u/ J0 c. ]
task = topic_model_training_tasks[user_uuid] ' j2 ?0 Q. M* p5 B* F6 ]' X( F if task.nmf_training_progress.status_code == 2: * a4 s) A0 D$ U6 t% _ w( L' n background_task.add_task(task.topic_keyword_optimization, tko_params)/ d( ^% O% S. K" h
return {"message": "已添加主题关键词优化后台任务"} " ~" k% f5 v/ ~- y, O; f& G else:0 Y( v" y: J$ m$ A3 O
raise HTTPException(status_code=404, detail="请先等待NMF训练结束!")% L' Q5 G# E; p" n* k6 z& s& _
else:( a% ^/ [' M% G3 A
raise HTTPException(status_code=404, detail="未找到相应的训练对象!") # G1 @- S# ~& h' y' D( I- b6 W 6 z9 v) z- m1 L. U1 a9 |. N* g. P5 u0 o$ B
@app.post(& u* O9 }0 A4 I
"/calculate/nmftraining/topicsplit/", % g3 M8 J$ o- M6 h! D7 _ tags=["用户交互"] 3 e9 c0 R8 C8 g$ o$ s; J1 q0 J)- P* Z1 C( E6 [4 I0 m
async def run_topic_split_task(ts_params: TopicSplitParams, + S4 F& @6 w- X background_task: BackgroundTasks, 4 P# U/ P4 S9 O1 w response: Response,, b3 m. N% q6 ^$ @# f
user_uuid: str = Cookie(None)): 9 D2 ]3 n! c/ d4 L S/ C7 G if not user_uuid: " @ a& p6 G4 \* E$ j user_uuid = create_user_uuid_and_set_cookie(response)( j* _, L" B( c3 n. Q5 q/ B4 p7 C& P
_% n$ [% O6 ]2 @( y$ x5 {& w1 c% p/ S global topic_model_training_tasks ( c$ J( x" H6 _. R% v6 | + @) q% w; \) p8 U9 }- V/ |7 V3 ] if user_uuid and user_uuid in topic_model_training_tasks: ( k2 n, j8 m4 k' J# g, o task = topic_model_training_tasks[user_uuid] 8 p* Y7 Z% e3 q# i5 z) }6 s- L5 x+ x8 { if task.nmf_training_progress.status_code == 2:( t. N) }3 y8 V# j1 M! J
background_task.add_task(task.topic_split, ts_params) ) o& A: A0 C+ O return {"message": "已添加主题拆分后台任务"}8 B- X+ {2 J: S
else:7 P/ E: U4 A3 ]# y
raise HTTPException(status_code=404, detail="请先等待NMF训练结束!")4 Q; I0 y1 y0 R3 j
else: 0 t' | J+ N0 T. S8 I' ` raise HTTPException(status_code=404, detail="未找到相应的训练对象!")! D3 I+ I4 T) D% C
0 D+ O0 V- c6 U$ {: N" E! p
+ K3 B/ p; M6 g; H3 l4 A@app.post( # ]( g% a2 R( s' r' |, L "/calculate/nmftraining/topicmerge/", / P( W K% B8 N5 w$ v tags=["用户交互"] - X2 P' P5 S1 | { L) : m; ]+ L4 k$ x" D. ~async def run_topic_merge_task(tm_params: TopicMergeParams,5 @1 p0 ~3 t8 b. ^3 E% ?. i
background_task: BackgroundTasks, ! g v9 `; I/ M response: Response,9 o( X+ G. t" {7 v; S+ V4 }
user_uuid: str = Cookie(None)):4 b: N2 z) j& I# V) E
if not user_uuid:' f' q* ?7 s8 G+ `
user_uuid = create_user_uuid_and_set_cookie(response) ; h& d4 L% ?" D) e) L( i 4 x) A# ]* h8 W. i! A+ K$ q global topic_model_training_tasks 3 C% Q8 b4 R Z' q Y9 @0 F+ u3 C( _" \4 Q1 @ if user_uuid and user_uuid in topic_model_training_tasks: h) C" G: M' U$ L( d) ^1 L, I
task = topic_model_training_tasks[user_uuid]( T% i J. `! `$ v2 a# q7 c
if task.nmf_training_progress.status_code == 2: 1 E4 l7 ^/ q& Q9 p* O- j! R background_task.add_task(task.topic_merge, tm_params) l4 D7 I/ V- [3 X% v6 i) y# _* q
return {"message": "已添加主题合并后台任务"}+ _- {2 c9 P( Z6 _5 A: j& v
else:2 W" c1 s+ x; b" _* B
raise HTTPException(status_code=404, detail="请先等待NMF训练结束!") [+ l$ |/ @) m4 v! [ else: T6 l' m8 v( C8 X. w: E
raise HTTPException(status_code=404, detail="未找到相应的训练对象!")' l" j) c# d) z7 Q% u
' U- s- ~ [0 `( `: W9 @) ~
3 n3 G! I6 a9 [7 D; G@app.post( 3 A5 P3 g) Y# @9 P. o( {+ C& q7 \ "/calculate/nmftraining/keywordinducedtopiccreate/", % ]7 t' W. B% B( }& U3 h& V tags=["用户交互"]0 J9 a: _7 A: U
)1 P; s, Z, m# q# g) o7 S
async def run_keyword_induced_topic_create_task(kitc_params: KeywordInducedTopicCreateParams, + k4 p; G$ N) S% _% r5 H# n4 f0 ? background_task: BackgroundTasks, . ?2 Z* g; Y0 x0 J) T& u response: Response, ( q; I: |8 ]9 @- a* v3 n user_uuid: str = Cookie(None)): . n( T$ x6 g& S+ ` U' h, P if not user_uuid: ' v j; w4 Y Q user_uuid = create_user_uuid_and_set_cookie(response) & ?# p6 w5 Y1 y( U: _5 [, {+ o) l7 A- a! G3 r* T% P
global topic_model_training_tasks , _% `( C0 R' n! L$ b+ A5 a t% S1 S R* b! b if user_uuid and user_uuid in topic_model_training_tasks:) ^4 o5 \9 m4 E- U
task = topic_model_training_tasks[user_uuid]: Z ^) m9 T+ ~9 h0 Z' m5 p" s) H
if task.nmf_training_progress.status_code == 2: ( C7 z: c# Q* \ f background_task.add_task(task.keyword_induced_topic_create, kitc_params) . Z3 B& @ c. L* Z; c- v& i return {"message": "已添加关键词诱导主题创建后台任务"}6 b. i8 K5 D) C- }
else: ! Z7 h. t0 Y8 M raise HTTPException(status_code=404, detail="请先等待NMF训练结束!"): F, M1 e8 m; H3 w
else:2 z' F5 c8 @- [* L# u& J+ f: i/ C
raise HTTPException(status_code=404, detail="未找到相应的训练对象!") + P+ `; P# {- Q% S6 j) ^3 w: f) V- {! D( T/ x, C1 @4 z4 { F
. f# S& Y/ Q/ s% j& L/ L
@app.post(" Z. R) t0 N7 F. ~+ l4 O, f
"/calculate/nmftraining/documentinducedtopiccreate/",, z$ B! H1 G; \
tags=["用户交互"]9 E$ F, Q" I! u9 j3 k
) ; G; {" g' x( uasync def run_document_induced_topic_create_task(ditc_params: DocumentInducedTopicCreateParams, 6 u X& B- B8 g5 Q9 o background_task: BackgroundTasks, / P# H5 K' M" d. j6 ~ response: Response,+ C, I3 B3 @2 ]5 c: x% l9 N7 R: ]
user_uuid: str = Cookie(None)): $ P& n/ v/ m7 L; q7 P& g if not user_uuid:1 o' X$ E V0 _, ~. x
user_uuid = create_user_uuid_and_set_cookie(response) . T. P5 H& d0 R$ J1 ~$ I- u5 t1 ` x$ [; W9 u
global topic_model_training_tasks6 V) H7 M- P" [+ ~9 y
f! ?, X$ A6 {5 r9 Z+ J' j e/ U if user_uuid and user_uuid in topic_model_training_tasks:& T: S d9 \6 C5 {& g
task = topic_model_training_tasks[user_uuid] }) q* B1 ~) o b( ` if task.nmf_training_progress.status_code == 2:# I! J- t; K! D4 \4 d- \
background_task.add_task(task.document_induced_topic_create, ditc_params) 5 H/ D8 i6 i' t9 W9 V" D* y4 v return {"message": "已添加文档诱导主题创建后台任务"} 1 p# b4 [8 j' i* b! x m else:) E8 q+ R7 w/ J: \8 I3 D
raise HTTPException(status_code=404, detail="请先等待NMF训练结束!"): A, l: x! J1 L" W2 q9 d
else: 4 v( Z0 S4 U R raise HTTPException(status_code=404, detail="未找到相应的训练对象!") & F- H$ p# {6 ^( \/ n, o 8 w& R# b! K. u8 a s% O. Q+ M: @2 Y8 h
@app.get( # s; y4 w$ z3 a* } "/calculate/nmftraining/newdoctopicdistributionpredict/",* G6 }, @& z4 V
tags=["主题预测"]9 c4 M3 C' w" p: ~5 Q: R
) 8 W! l9 q8 v- U; Q: v7 uasync def run_new_document_topic_distribution_predict_task() p& U3 P5 X3 N& K' G8 k
new_doc_text: str,5 C* p! j3 M6 [4 R4 D3 X
background_task: BackgroundTasks, ' P5 w' l4 K* ~4 A response: Response,+ J' P! S% V# q* k [ r2 e( W
user_uuid: str = Cookie(None)): % ~7 H, ~: D0 c6 z. p if not user_uuid:: r3 p! {+ a# y6 n( i0 ^0 ^
user_uuid = create_user_uuid_and_set_cookie(response) $ g0 O$ F! i' ~ 5 J$ B: O J% C4 G3 N global topic_model_training_tasks: F9 n2 N" ^& f& _ f, B
" i9 T0 C3 H/ J0 {' b4 q; K if user_uuid and user_uuid in topic_model_training_tasks: ( \7 Q9 Y4 p1 @+ ~ e+ u task = topic_model_training_tasks[user_uuid] : g' l1 f( f7 ~* g- i if task.nmf_training_progress.status_code == 2:6 N, s h P6 o* M3 H* X, d R
background_task.add_task(6 \: ~# j, R4 Z9 `, f
task.new_doc_topic_distribution_predict, * P& _4 f# H: F' ^1 t6 A0 Q8 A, Y new_doc_text 9 W. C# b% c3 U" O2 n5 D ) ) m; d5 ~1 ]: H O8 Y5 o return {"message": "已添加新文档主题分布预测后台任务"}# A8 T8 @# _ J, `( U1 f
else: 4 O& h# F, }& J1 Y4 ?3 e" l1 ] raise HTTPException(status_code=404, detail="请先等待NMF训练结束!") 9 z# u: A2 c# A else:7 L! s5 S; R3 z! k+ |0 w5 ]4 m' z
raise HTTPException(status_code=404, detail="未找到相应的训练对象!") # V; w$ @) R! a! X5 j5 E9 z! }/ [6 t $ Z1 A# |" g4 [0 W d) { . Y6 `8 o& b% I6 j' L9 D@app.get( 6 b7 M F! i( l* ?) b8 u "/calculate/preprocessing/progress/", 2 G- M7 H$ ]& J# v- t! ~/ S7 `0 {* [ tags=["计算模块数据接口"], . s: @+ G* o: {/ H' {7 f4 b. A response_model=TextPreprocessingProgress- k6 J% D! @% h0 l
) : X M4 O/ X5 ~3 o" hdef get_text_preprocessing_task_progress(user_uuid: str = Cookie(None)):# m! D$ y, G q2 E
global topic_model_training_tasks # o% j/ M- S5 B. H! ~/ o9 ?: g1 @9 q
if (not user_uuid) or (user_uuid not in topic_model_training_tasks): ' q0 T& U" ?2 W* O$ m* k. A; f raise HTTPException(status_code=404, detail="未找到相应的训练对象!")) R+ A8 Q; _+ m5 V
return topic_model_training_tasks[user_uuid].text_preprocessing_progress ( I8 G, C' N- v" h: C9 w1 D/ Y& D 3 F5 J0 \0 \: w/ ~5 z % m5 k) F0 D+ a/ w9 k3 D1 v8 O6 G@app.get(7 d( g9 c1 l" B: S9 R, e0 l! b
"/calculate/nmftraining/progress/", ( d0 s7 w0 A0 S1 X, ?( \' h" ] tags=["计算模块数据接口"],2 Z& b6 H4 O% A
response_model=NMFTrainingProgress$ y( p: U& ]! L, l8 `
) & G$ W6 P& W4 x- W+ ?6 kdef get_nmf_training_task_progress(user_uuid: str = Cookie(None)): + e0 E1 |' ~* O: Y1 W global topic_model_training_tasks , ?3 [3 T; ~$ _2 d! Q$ Q 9 F6 s" u: g- ^ | if (not user_uuid) or (user_uuid not in topic_model_training_tasks): 4 I) k% |/ K( F8 `1 Q6 P* {. Q raise HTTPException(status_code=404, detail="未找到相应的训练对象!") / x2 Y' v! @% w0 R) S+ _3 O/ d( b' Y+ |
task = topic_model_training_tasks[user_uuid]0 U4 P7 V6 k* R; O( q- b
if task.text_preprocessing_progress.status_code == 2:1 q R& R/ d8 H0 J& _
return task.nmf_training_progress 0 d; ]7 Z$ [: B: @6 [4 @* E else: " Y/ b4 ^; ^1 }2 V( t raise HTTPException(status_code=404, detail="请先进行文本预处理任务!") 8 v- s4 C4 A1 `, h* h7 t7 G 0 T6 r! q& F% u6 B7 t, W5 Y& y# f+ g; A9 k
@app.get( # \$ f! k4 q+ C' V U "/calculate/umap/progress/", ) F! q" `- z- O tags=["计算模块数据接口"], 3 v0 }0 _5 o+ c N1 I7 g+ Z response_model=UMAPProgress " ^0 ?* B& Q! u/ E b) 3 s' W0 u/ }9 l. }def get_umap_task_progress(user_uuid: str = Cookie(None)): , S1 J: H- i* U( U; k) t global topic_model_training_tasks, V2 K$ s8 o% _7 e# _; b
; ~8 U% b6 C' D2 T& E, Y
if (not user_uuid) or (user_uuid not in topic_model_training_tasks):% t. |9 G1 i3 y
raise HTTPException(status_code=404, detail="未找到相应的训练对象!") % }! a& V4 g; i! \ , m6 n6 {8 v& F! d: q task = topic_model_training_tasks[user_uuid]$ e! M: d/ e# d N3 Q* c x' R
if task.nmf_training_progress.status_code == 2:0 f" P* J" M+ A( }( h/ \
return task.umap_progress 5 s1 h2 {- i$ N1 T# @ else: & i+ P& d6 W# b r! A0 ], Y raise HTTPException(status_code=404, detail="请先等待NMF训练结束!") / e% i( R0 O* e" D4 m; d, g* b9 z. r7 ? D7 \- K# t
; k, ~( ?' M9 v
@app.get( . `/ }5 W% b* R8 t4 Y2 ?% Z "/calculate/nmftraining/predict/progress/",6 f3 c5 m) I4 u5 c8 x) [
tags=["计算模块数据接口"], 0 Y! B$ N- b6 f7 r response_model=NewDocTopicDistributionPredictProgress& A- N* `7 M, E# _7 n7 Y0 O! ?) F
) 4 c8 l1 I2 S; Ydef get_new_doc_topic_distribution_predict_task_progress(user_uuid: str = Cookie(None)): * u; o- y) \( i2 n t! a0 D6 z. U global topic_model_training_tasks7 }% s# n9 D& @7 i" L) \
" v. _! a. k1 D# ^$ @! s5 x# b8 c
if (not user_uuid) or (user_uuid not in topic_model_training_tasks):/ q2 t, i$ x9 I, f9 x/ O e9 J) P
raise HTTPException(status_code=404, detail="未找到相应的训练对象!") & O' r1 P. e4 }7 o3 t* y4 O; H, `3 ]4 x$ ? U/ v
task = topic_model_training_tasks[user_uuid] 2 c4 t, P# j7 `% G1 E% _/ w; L if task.nmf_training_progress.status_code == 2: + l2 z5 O L! S& R" g9 s5 l return task.topic_distribution_predict_progress 0 y) \. M5 G; V$ R2 G, d2 B else: + o, b8 s. @$ L- c6 q1 ~% | raise HTTPException(status_code=404, detail="请先等待NMF训练结束!")# u' s, Q0 i$ ~
: l1 S& O7 Y2 @
: ~) T1 Z) L5 i, E+ F. G
@app.get(1 ]4 t1 k: p+ Y5 a) g8 w% ?) [
"/calculate/details/document/", 1 U! F4 ~5 f) C8 x5 L& u tags=["计算模块数据接口"], 1 O8 C" Y3 o( C/ k, x% S response_model=DocumentDetails$ X2 }& ]$ b: C7 p1 Q% j9 V
) ; Y5 x. _0 o% k$ P9 z' o% adef get_document_details(doc_id: int, user_uuid: str = Cookie(None)):- ~ \: X' }* I# q5 _
global topic_model_training_tasks - t9 H" ]8 s; m # h4 @! u( C% |. m7 y if (not user_uuid) or (user_uuid not in topic_model_training_tasks): % ?8 G8 U6 o: M raise HTTPException(status_code=404, detail="未找到相应的训练对象!") , e, b8 q9 D! [, i; Q( a3 t. ]) J! s+ s z4 }$ u
task = topic_model_training_tasks[user_uuid] 3 J4 s! r0 q; K: S: I; d* c6 K if task.nmf_training_progress.status_code == 2: $ _$ C- Z# G6 P) i. S return task.get_document_details(doc_id) / s3 D' T: ]% @$ [ f$ D$ ~+ C else: & W8 N" D# @0 z2 D- F9 G' P raise HTTPException(status_code=404, detail="请先等待NMF训练结束!") # {* Z( g3 v J; J# {& R& }- i+ s) P : `% ^, D4 A/ T5 k8 q% x# D' v0 P& i1 }3 u1 d
@app.get(' p0 Q, z1 Y z2 o& p$ A% @
"/calculate/details/topic/", % ?, ^) }+ ~3 j: i7 l3 c) x tags=["计算模块数据接口"], 3 @. J% J1 ~6 p5 u: P response_model=TopicDetails$ s9 Y, ]# K) I) S
) 4 {( ^; k4 f2 Hdef get_topic_details(topic_id: int, user_uuid: str = Cookie(None)): 8 j4 }# M* D8 s8 O V* k global topic_model_training_tasks 0 h( v" d. K5 W' k1 o$ ^$ Z3 x: R( P! z7 x/ F9 \- w4 h" R
if (not user_uuid) or (user_uuid not in topic_model_training_tasks):% |' O3 R% ~" p0 s
raise HTTPException(status_code=404, detail="未找到相应的训练对象!"), C( G. `" z# E4 P, B! F5 h( R/ ~' F
! n/ {) Q$ i, I# G \/ f3 w7 b task = topic_model_training_tasks[user_uuid] 0 y9 o) j$ ?# c. n6 Y ^8 T if task.nmf_training_progress.status_code == 2:) t7 L. d6 t" U1 [3 ]% K/ y0 ?
return task.get_topic_details(topic_id)( L( i- N+ m" P' ~
else: : t! V1 u3 u' r9 s raise HTTPException(status_code=404, detail="请先等待NMF训练结束!") ' j1 s+ ^5 s4 `5 h. e5 U# \6 J% c1 g; C: W2 ]; }1 M
4 y9 Y, L9 L, L1 q6 Q6 K@app.get( 2 ?6 g* e) i& W' r "/calculate/userinteraction/info/keyword/",; ]3 r! b+ v2 G) N& F1 ?
tags=["计算模块数据接口"], . A) \1 J; W; v' P; [4 C response_model=TopicKeywordInfo : I% j) g+ n& W8 B. J% J)$ c1 w3 z/ z( t0 _/ b# V
def get_topic_keyword_info(topic_id: int, user_uuid: str = Cookie(None)): + f* N5 p- o; U/ [- y8 c& s global topic_model_training_tasks/ {, Z& w* w% q( H* u
+ F' c% A0 D/ C" M& \; ?$ T1 z
if (not user_uuid) or (user_uuid not in topic_model_training_tasks):+ t: d: W* \8 B/ t5 \2 g& N" D
raise HTTPException(status_code=404, detail="未找到相应的训练对象!")1 u0 t3 q4 ]3 d- e' m5 ?# k
1 Z. R! P- i0 [2 U; u2 Y7 n task = topic_model_training_tasks[user_uuid] 6 o y5 t2 P) I" w. a k if task.nmf_training_progress.status_code == 2: }( w& J' B" p `: k return task.get_topic_keyword_info(topic_id)5 r0 j' p$ y( W
else: d w1 n5 P% M* v- X raise HTTPException(status_code=404, detail="请先等待NMF训练结束!") ; v" X7 L. }- C1 S! p1 D 7 U/ W: C P4 \( N# ] v* s' m: v, o. y2 w3 l! [+ o6 H$ p
@app.get(. M( C* T' |- I3 S% }4 H- W
"/calculate/userinteraction/info/topicmergekeyword/",1 v& t' W" N% I
tags=["计算模块数据接口"],& F+ K8 t( a8 N6 Y" \
response_model=TopicMergeKeywordInfo9 ]! @- t7 _7 Y
) ' q5 }% h6 Y) B9 F* v" G/ b7 ~def get_topic_merge_keyword_info(topic1_id: int, topic2_id: int, user_uuid: str = Cookie(None)):5 ^$ a3 I& R! l/ F
global topic_model_training_tasks, C: a, h6 C# T3 b' w7 u4 Q
- ], J& X6 p1 W# R if (not user_uuid) or (user_uuid not in topic_model_training_tasks):% I- N2 \% t8 J( y% P
raise HTTPException(status_code=404, detail="未找到相应的训练对象!")7 C$ ~, H' p9 [5 Z