Commit c8b59522 authored by Davis King's avatar Davis King

refined spec a little

parent d3c26551
......@@ -570,7 +570,7 @@ namespace dlib
// ----------------------------------------------------------------------------------------
template <typename T, typename alloc>
std::pair<T,T> learn_platt_scaling (
std::pair<double,double> learn_platt_scaling (
const std::vector<T,alloc>& scores,
const std::vector<T,alloc>& labels
)
......
......@@ -132,7 +132,7 @@ namespace dlib
// ----------------------------------------------------------------------------------------
// ----------------------------------------------------------------------------------------
inline double platt_scale (
double platt_scale (
const std::pair<double,double>& params,
const double score
);
......@@ -144,7 +144,7 @@ namespace dlib
// ----------------------------------------------------------------------------------------
template <typename T, typename alloc>
std::pair<T,T> learn_platt_scaling (
std::pair<double,double> learn_platt_scaling (
const std::vector<T,alloc>& scores,
const std::vector<T,alloc>& labels
);
......@@ -154,10 +154,12 @@ namespace dlib
- is_binary_classification_problem(scores,labels) == true
ensures
- This function learns to map scalar values into well calibrated probabilities
using Platt scaling. In particular, it returns a params object such that:
- for all valid i:
- platt_scale(params,scores[i]) == the scaled version of the scalar value
scores[i]. That is, the output is a number between 0 and 1.
using Platt scaling. In particular, it returns a params object such that,
for all valid i:
- platt_scale(params,scores[i]) == the scaled version of the scalar value
scores[i]. That is, the output is a number between 0 and 1. In
particular, platt_scale(params,scores[i]) is meant to represent the
probability that labels[i] == +1.
- This function is an implementation of the algorithm described in the following
papers:
Probabilistic Outputs for Support Vector Machines and Comparisons to
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment