sklearn.LabelEncoder with never seen before values

后端 未结 12 1011
执笔经年
执笔经年 2020-11-27 10:37

If a sklearn.LabelEncoder has been fitted on a training set, it might break if it encounters new values when used on a test set.

The only solution I c

12条回答
  •  忘掉有多难
    2020-11-27 11:34

    If it is just about training and testing a model, why not just labelencode on entire dataset. And then use the generated classes from the encoder object.

    encoder = LabelEncoder()
    encoder.fit_transform(df["label"])
    train_y = encoder.transform(train_y)
    test_y = encoder.transform(test_y)
    

提交回复
热议问题