时间戳

首页 > 解决方案 > 当我们拆分训练和测试时，我们希望将目标变量设为动态

python - 当我们拆分训练和测试时，我们希望将目标变量设为动态

问题描述

当我们尝试使用survived作为目标变量进行拆分时，它工作正常。但是我们动态需要它，没有目标变量名。以下是我们尝试的示例：

from sklearn.model_selection import train_test_split     
def train_test_split(d):
    target='Survived'
    x=d.drop(target,axis=1)
    y=d.target
    x_train,x_test,y_train,y_test=train_test_split(x,y,
                                        test_size=0.33)

非常感谢您的帮助！

标签： pythonmachine-learning

解决方案

我希望我正确理解了你的问题。

如何制作target一个需要像这样传递给函数的参数：

   def train_test_split(d, target):
       x=d.drop(target,axis=1)
       y=d.target
       x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.33)

然后，您可以传递您需要的任何目标值，例如train_test_split(d, "Died")

推荐阅读