在机器学习中我们需要将数据“喂”给计算机,计算机根据我们的算法来对数据进行学习,从而具备预测能力,对其他的数据进行预测。
在这里我们喂给模型的的数据就是我们的训练集,模型用来预测的数据叫做测试集用来评判模型的预测能力。

sklearn中分割训练集和测试集

  • 使用sklearn中的train_test_split对分割训练集和测试集