Plotting a ROC curve in scikit yields only 3 points

前端 未结 4 828
太阳男子
太阳男子 2020-12-31 03:06

TLDR: scikit\'s roc_curve function is only returning 3 points for a certain dataset. Why could this be, and how do we control how many points to get ba

4条回答
  •  一整个雨季
    2020-12-31 03:39

    It's not necessary to get 1 point except (0,0) and (1,1). I'm using mushrooms dataset from kaggle for a binary classification problem. Procuring fpr and tpr from roc_curve, I'm getting 4 more points, though their value is more or less same.

    fpr = {0, 0, 0.02290076, 0.0267176, 0.832061, 1}

    tpr = {0, 0.0315361, 0.985758, 0.996948, 1, 1}

    I'm not sure if we can consider this as 1 point because plotting the curve using this looks like the one shown in question.

提交回复
热议问题