Welcome to WuJiGu Developer Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
4.0k views
in Technique[技术] by (71.8m points)

machine learning - Maximise custom function for weighted binary classification

I have the following problem: Given two classes A and B which should be predicted using Sklearn or XGBoost. The function which has to be maximised is f(x) = A'^2/(A'+B'). Meaning that A' is the number of passing As from the dataset and B' the number of passing Bs. So the cost for removing a A is higher than removing a B. (Maximised should be TruePositive^2/(TruePositive+FalsePositive) Defining a custom evaluation metric is not the problem, my problem is that I do not know which objective function/loss function (sorry, not sure what is the correct label is) I should use so that the model explicitly trains to maximise TP^2/(TP+FP) and not train sth. different.

Thanks in advance!


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Answer

0 votes
by (71.8m points)
等待大神答复

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome to WuJiGu Developer Q&A Community for programmer and developer-Open, Learning and Share
...