You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I am new comer of TransmogrifAI and working on running Titanic Binary Classification according to the link. Currently I try to choose the best model by auROC. I revise the example's code as follows. val evaluator = Evaluators.BinaryClassification.auROC().setLabelCol(survived). setPredictionCol(prediction)
but don't achieve it. The model is still chosen by auPR.
Could someone help tell me how to do it? Thanks a lot.
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
I am new comer of TransmogrifAI and working on running Titanic Binary Classification according to the link. Currently I try to choose the best model by auROC. I revise the example's code as follows.
val evaluator = Evaluators.BinaryClassification.auROC().setLabelCol(survived). setPredictionCol(prediction)
but don't achieve it. The model is still chosen by auPR.
Could someone help tell me how to do it? Thanks a lot.
Beta Was this translation helpful? Give feedback.
All reactions