【问题标题】:To change the output class label value of a predict function in OneclassSVM更改 OneclassSVM 中预测函数的输出类标签值
【发布时间】:2021-09-13 07:30:41
【问题描述】:

当我使用 OneClassSVM 时,我们确认 estimator.predict (X_test) 得到的结果分别为 1 和 -1。每个都意味着一个异常值和一个内部值。但我想要的是用不同的值来标记它,比如 0,1 而不是 -1,1。我以为我可以给出一个具体的论据来预测这样做,但我找不到我想要的搜索结果。

from sklearn import OneClassSVM

check = OneClassSVM(kernel='rbf', gamma='scale')
check.fit(X_train, y_train)
check.predict(X_test)

我使用了上面的代码。

【问题讨论】:

    标签: scikit-learn sklearn-pandas


    【解决方案1】:

    没有指定标签的内置函数。但是,您可以使用np.where() 执行此操作:

    import numpy as np
    pred = np.array([-1, 1, -1, 1])
    
    np.where(pred==-1, 'outlier_value', 'internal_value')
    

    输出:

    array(['outlier_value', 'internal_value', 'outlier_value',
       'internal_value'], dtype='<U14')
    

    【讨论】:

      猜你喜欢
      • 2021-01-02
      • 2015-02-01
      • 2022-06-14
      • 1970-01-01
      • 2017-09-15
      • 2020-05-11
      • 2021-09-14
      • 2020-09-11
      • 1970-01-01
      相关资源
      最近更新 更多