hong

model/yuqing.py

@@ -0,0 +1,48 @@
+import pandas as pd  # 用于数据处理
+import numpy as np  # 用于科学计算
+import csv  # 用于读取CSV文件
+from snownlp import SnowNLP  # 用于中文自然语言处理（此处未实际使用）
+from sklearn.feature_extraction.text import TfidfVectorizer  # 用于文本特征提取
+from sklearn.naive_bayes import MultinomialNB  # 用于多项式朴素贝叶斯分类
+from sklearn.model_selection import train_test_split  # 用于划分训练集和测试集
+from sklearn.metrics import accuracy_score  # 用于计算模型准确度
+
+
+def getSentiment_data():
+    # 从CSV文件中读取情感数据
+    sentiment_data = []
+    with open('./target.csv', 'r', encoding='utf8') as readerFile:
+        reader = csv.reader(readerFile)
+        for data in reader:
+            sentiment_data.append(data)
+    return sentiment_data
+
+
+ef
+model_train():
+# 获取情感数据并转换为DataFrame
+sentiment_data = getSentiment_data()
+df = pd.DataFrame(sentiment_data, columns=['text', 'sentiment'])
+
+# 将数据集划分为训练集和测试集，测试集占20%
+train_data, test_data = train_test_split(df, test_size=0.2, random_state=42)
+
+# 初始化TfidfVectorizer，并对训练集和测试集进行文本特征提取
+vectorize = TfidfVectorizer()
+X_train = vectorize.fit_transform(train_data['text'])
+y_train = train_data['sentiment']
+X_test = vectorize.transform(test_data['text'])
+y_test = test_data['sentiment']
+
+# 初始化多项式朴素贝叶斯分类器，并进行训练
+classifier = MultinomialNB()
+classifier.fit(X_train, y_train)
+
+# 对测试集进行预测
+y_pred = classifier.predict(X_test)
+
+# 计算模型准确度
+accuracy = accuracy_score(y_test, y_pred)
+
+if __name__ == "__main__":
+    model_train()  # 训练模型并计算准确度