-
Notifications
You must be signed in to change notification settings - Fork 4
/
main.cpp
37 lines (30 loc) · 1.32 KB
/
main.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
// Copyright (c) 2014 Vassileios Balntas
// balntas@robots.ox.ac.uk
// cppROC is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#include "roc.hpp"
#include <iostream>
// example on how to read a file with classification data
// and get the ROC curve
int main(int argc, char *argv[])
{
// format of the file:
// column 0 -> classification scores (higher is for positive class)
// column 1 -> class labels (1 is positive class)
ROC roc("classification_result_example.txt");
// the format of the output file is
// column 0 -> False positive points in curve
// column 1 -> True positive points in curve
roc.writeToFile("roc.txt");
// roc.txt can then be plotted with gnuplot, mathGL etc.
std::cout << roc.getAreaUnderCurve()<<'\n';
return 0;
}