更改 Sklearn 进行分类类

Question

将 sklearn 中的 make_classification 用于二元分类问题，我得到标签 0 和 1。无论如何，我可以将其更改为 -1 而不是 0 吗？

Answer 1

您可以使用 np.where :

from sklearn.datasets import make_classification
import numpy as np

X,y = make_classification()
y_transformed = np.where(y==0,-1,1)

Change Sklearn Make Classification Classes