问答 · ANSWER

数据集准备及划分是什么?

直接回答
数据集准备指将数据整理为特征矩阵X和标签向量y的标准格式。划分则是将数据分为训练集和测试集,以支持对模型性能进行客观评估。

关键要点

  • 数据集标准格式包括特征矩阵X(形状为n_samples×n_features)和标签向量y(形状为n_samples)。
  • 划分数据集是为了避免在训练集上测试导致的评估偏差,实现客观评估。
  • 常用train_test_split函数划分,参数test_size指定测试集比例,random_state支持结果可复现。
  • 完整建模流程为:准备数据、划分数据、训练模型、进行预测、评估性能。

怎么选 / 怎么做

1. 准备数据:将数据整理为特征矩阵X和标签向量y的标准格式。
2. 划分数据:使用train_test_split函数,指定test_size(如0.25)和random_state参数,得到X_train, X_test, y_train, y_test。
3. 训练模型:在训练集上调用模型的.fit(X_train, y_train)方法。
4. 进行预测:在测试集上调用模型的.predict(X_test)方法得到预测结果y_pred。
5. 评估性能:比较y_pred和y_test,使用准确率等指标评估。

常见问题

为什么需要划分数据集?

避免用练习题当考试,支持模型评估的客观性。

数据集的标准格式是什么?

特征矩阵X(二维数组)和标签向量y(一维数组)。

train_test_split的random_state有什么用?

设置随机种子,便于结果复现和实验比较。

深入学

小象教研组整理 · 内容来自课程讲义与企业内训素材,仅供学习参考。