Commit d3a3288e authored by Davis King's avatar Davis King

Added a max iterations option to the sequence labeling and segmentation

learning tools.
parent 28da58f0
......@@ -76,6 +76,16 @@ namespace dlib
double get_epsilon (
) const { return eps; }
unsigned long get_max_iterations (
) const { return max_iterations; }
void set_max_iterations (
unsigned long max_iter
)
{
max_iterations = max_iter;
}
void set_max_cache_size (
unsigned long max_size
)
......@@ -215,6 +225,7 @@ namespace dlib
prob.be_verbose();
prob.set_epsilon(eps);
prob.set_max_iterations(max_iterations);
prob.set_c(C);
prob.set_max_cache_size(max_cache_size);
for (unsigned long i = 0; i < loss_values.size(); ++i)
......@@ -230,6 +241,7 @@ namespace dlib
double C;
oca solver;
double eps;
unsigned long max_iterations;
bool verbose;
unsigned long num_threads;
unsigned long max_cache_size;
......@@ -240,6 +252,7 @@ namespace dlib
C = 100;
verbose = false;
eps = 0.1;
max_iterations = 10000;
num_threads = 2;
max_cache_size = 5;
loss_values.assign(num_labels(), 1);
......
......@@ -47,6 +47,7 @@ namespace dlib
- #get_c() == 100
- this object isn't verbose
- #get_epsilon() == 0.1
- #get_max_iterations() == 10000
- #get_num_threads() == 2
- #get_max_cache_size() == 5
- #get_feature_extractor() == a default initialized feature_extractor
......@@ -60,6 +61,7 @@ namespace dlib
- #get_c() == 100
- this object isn't verbose
- #get_epsilon() == 0.1
- #get_max_iterations() == 10000
- #get_num_threads() == 2
- #get_max_cache_size() == 5
- #get_feature_extractor() == fe
......@@ -119,6 +121,22 @@ namespace dlib
training sample is within epsilon of its optimal value".
!*/
void set_max_iterations (
unsigned long max_iter
);
/*!
ensures
- #get_max_iterations() == max_iter
!*/
unsigned long get_max_iterations (
);
/*!
ensures
- returns the maximum number of iterations the SVM optimizer is allowed to
run before it is required to stop and return a result.
!*/
void set_max_cache_size (
unsigned long max_size
);
......
......@@ -72,6 +72,16 @@ namespace dlib
double get_epsilon (
) const { return trainer.get_epsilon(); }
unsigned long get_max_iterations (
) const { return trainer.get_max_iterations(); }
void set_max_iterations (
unsigned long max_iter
)
{
trainer.set_max_iterations(max_iter);
}
void set_max_cache_size (
unsigned long max_size
)
......
......@@ -44,6 +44,7 @@ namespace dlib
- #get_c() == 100
- this object isn't verbose
- #get_epsilon() == 0.1
- #get_max_iterations() == 10000
- #get_num_threads() == 2
- #get_max_cache_size() == 40
- #get_feature_extractor() == a default initialized feature_extractor
......@@ -59,6 +60,7 @@ namespace dlib
- #get_c() == 100
- this object isn't verbose
- #get_epsilon() == 0.1
- #get_max_iterations() == 10000
- #get_num_threads() == 2
- #get_max_cache_size() == 40
- #get_feature_extractor() == fe
......@@ -111,6 +113,22 @@ namespace dlib
per training sample is within epsilon of its optimal value".
!*/
void set_max_iterations (
unsigned long max_iter
);
/*!
ensures
- #get_max_iterations() == max_iter
!*/
unsigned long get_max_iterations (
);
/*!
ensures
- returns the maximum number of iterations the SVM optimizer is allowed to
run before it is required to stop and return a result.
!*/
void set_max_cache_size (
unsigned long max_size
);
......
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