panda/bettafish-company
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

【trainModel.py】修改使文件可以运行

Browse Source
This commit is contained in:
redhongx
2024-07-04 12:44:10 +08:00
parent a1d1e8d4a6
commit a5280290b3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/trainModel.py
+1 -1
View File
@@ -1,7 +1,7 @@
import pandas as pd # 用于数据处理 import pandas as pd # 用于数据处理
import numpy as np # 用于科学计算 import numpy as np # 用于科学计算
import csv # 用于读取CSV文件 import csv # 用于读取CSV文件
from snownlp import SnowNLP # 用于中文自然语言处理（此处未实际使用） # from utils.mynlp import SnowNLP # 用于中文自然语言处理（此处未实际使用）
from sklearn.feature_extraction.text import TfidfVectorizer # 用于文本特征提取 from sklearn.feature_extraction.text import TfidfVectorizer # 用于文本特征提取
from sklearn.naive_bayes import MultinomialNB # 用于多项式朴素贝叶斯分类 from sklearn.naive_bayes import MultinomialNB # 用于多项式朴素贝叶斯分类
from sklearn.model_selection import train_test_split # 用于划分训练集和测试集 from sklearn.model_selection import train_test_split # 用于划分训练集和测试集