提问:请写一个使用逻辑回归模型判断用户更喜欢文心一言还是chatGPT的训练代码
时间: 2023-05-16 13:07:49 浏览: 148
利用文心一言和CHATGPT写新闻报道
5星 · 资源好评率100%
以下是使用逻辑回归模型判断用户更喜欢文心一言还是ChatGPT的训练代码:
```python
import pandas as pd
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# 读取数据
data = pd.read_csv('data.csv')
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(data.drop('label', axis=1), data['label'], test_size=0.2, random_state=42)
# 训练逻辑回归模型
lr = LogisticRegression()
lr.fit(X_train, y_train)
# 预测测试集
y_pred = lr.predict(X_test)
# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print('Accuracy:', accuracy)
```
其中,`data.csv`是包含用户对文心一言和ChatGPT的喜好评分的数据集,`label`列表示用户更喜欢哪个。训练集和测试集的划分比例为8:2。最终输出的是模型在测试集上的准确率。
注意:这只是一个示例代码,实际应用中需要根据具体情况进行调整和优化。
阅读全文