ChatGPT生成文本检测器算法挑战大

比赛链接：2023 iFLYTEK A.I.开发者大赛-讯飞开放平台 (xfyun.cn)

1、数据加载和预处理

import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split, cross_val_predict
from sklearn.linear_model import LogisticRegression
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics import classification_report
from sklearn.pipeline import Pipeline
from sklearn.model_selection import GridSearchCV

# 加载数据
train_data = pd.read_csv('ChatGPT生成文本检测器公开数据-更新/train.csv')
test_data = pd.read_csv('ChatGPT生成文本检测器公开数据-更新/test.csv')

# 数据预处理
train_data['content'] = train_data['content'].apply(lambda x: x[1:-1])
test_data['content'] = test_data['content'].apply(lambda x: x[1:-1])

2、训练集和验证集划分

# 划分训练集和验证集
train_text, valid_text, train_label, valid_label = train_test_split(
    train_data['content'], train_data['label'], test_size=0.2, random_state=42
)

3、Pipeline构建

使用Pipeline构建了一个包含TF-IDF向量化和逻辑回归模型的流水线。

# 使用Pipeline进行流水线构建
pipeline = Pipeline([
    ('tfidf', TfidfVectorizer(token_pattern=r'\w{1,}', max_features=5000, ngram_range=(1, 2))),
    ('model', LogisticRegression(max_iter=1000))
])

4、网格搜索参数

定义了要进行网格搜索的参数范围，其中包括TF-IDF的最大特征数和逻辑回归的正则化参数C。

# 网格搜索参数
param_grid = {
    'tfidf__max_features': [2000, 5000],
    'model__C': [0.1, 1, 10]
}

5、网格搜索交叉验证

使用GridSearchCV在训练集上执行网格搜索交叉验证，寻找最佳参数组合。

# 在训练集上进行网格搜索交叉验证
grid_search = GridSearchCV(pipeline, param_grid, cv=3, scoring='f1_macro')
grid_search.fit(train_text, train_label)

6、最佳模型获取

从网格搜索结果中选择出最佳模型。

# 最佳模型
best_model = grid_search.best_estimator_

7、预测和评估

使用最佳模型预测验证集的标签，并通过classification_report打印验证集的分类报告。

# 打印最佳参数
print("Best Parameters:", grid_search.best_params_)

# 预测验证集
valid_predictions = best_model.predict(valid_text)

# 打印验证集的分类报告
print("Validation Set Classification Report:")
print(classification_report(valid_label, valid_predictions))

8、测试集预测和保存

使用最佳模型对测试集进行预测，并将预测结果保存到CSV文件中。

# 使用最佳模型进行测试集预测
test_predictions = best_model.predict(test_data['content'])
test_data['label'] = test_predictions

# 保存预测结果
test_data[['name', 'label']].to_csv('tfidf_predictions.csv', index=None)