我正在使用 train_test_split
来训练模型并使用 predict
检查结果。如何继续预测其他数据的标签,例如来自测试集或用户输入的标签?
最佳答案
您可以使用自定义数据进行预测,只要它具有与您的训练数据相同的特征数量、顺序和类型,并将其放入 array
类型,而不是 list
。如果满足这些条件,您可以将该数组发送到模型,以使用普通的 predict()
方法进行预测。
关于python - 使用决策树进行自定义数据预测,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57147986/