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

Question

当我们尝试使用 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)

非常感谢您的帮助!

Answer 1

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

如何将 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")