October 2017
Intermediate to advanced
1159 pages
26h 10m
English
In this section, we will implement our own baseline model by following the approach that the KDD Cup organizers took. However, before we go to the model, let's first implement the evaluation engine that will return AUC on all three problems.
Now, let's take a closer look at the evaluation function. The evaluation function accepts an initialized model, cross-validates the model on all three problems, and reports the results as an area under the ROC curve (AUC), as follows:
public static double[] evaluate(Classifier model) throws Exception { double results[] = new double[4]; String[] labelFiles = new String[]{ "churn", "appetency", "upselling"}; double overallScore = 0.0; for (int i = 0; i < labelFiles.length; i++) ...