Commit 79656869 authored by Davis King's avatar Davis King

Simplified CLI code

parent 25bf3c51
...@@ -12,8 +12,8 @@ ...@@ -12,8 +12,8 @@
#include "dlib/matrix.h" #include "dlib/matrix.h"
typedef dlib::cmd_line_parser<char>::check_1a_c clp;
using dlib::command_line_parser;
typedef std::map<unsigned long, double> sparse_sample_type; typedef std::map<unsigned long, double> sparse_sample_type;
typedef dlib::matrix<sparse_sample_type::value_type::second_type,0,1> dense_sample_type; typedef dlib::matrix<sparse_sample_type::value_type::second_type,0,1> dense_sample_type;
......
...@@ -16,6 +16,7 @@ ...@@ -16,6 +16,7 @@
#include <vector> #include <vector>
#include "dlib/cmd_line_parser.h"
#include "dlib/data_io.h" #include "dlib/data_io.h"
#include "dlib/svm.h" #include "dlib/svm.h"
...@@ -24,7 +25,7 @@ using namespace dlib; ...@@ -24,7 +25,7 @@ using namespace dlib;
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
static void static void
parse_args (clp& parser, int argc, char* argv[]) parse_args (command_line_parser& parser, int argc, char* argv[])
{ {
try { try {
// Algorithm-independent options // Algorithm-independent options
...@@ -107,18 +108,16 @@ parse_args (clp& parser, int argc, char* argv[]) ...@@ -107,18 +108,16 @@ parse_args (clp& parser, int argc, char* argv[])
int int
main (int argc, char* argv[]) main (int argc, char* argv[])
{ {
clp parser; command_line_parser parser;
parse_args(parser, argc, argv); parse_args(parser, argc, argv);
const clp::option_type& option_alg = parser.option("a");
const clp::option_type& option_in = parser.option("in");
std::vector<sparse_sample_type> sparse_samples; std::vector<sparse_sample_type> sparse_samples;
std::vector<double> labels; std::vector<double> labels;
load_libsvm_formatted_data ( load_libsvm_formatted_data (
option_in.argument(), parser.option("in").argument(),
sparse_samples, sparse_samples,
labels labels
); );
...@@ -155,6 +154,7 @@ main (int argc, char* argv[]) ...@@ -155,6 +154,7 @@ main (int argc, char* argv[])
// Randomize the order of the samples, labels // Randomize the order of the samples, labels
randomize_samples (dense_samples, labels); randomize_samples (dense_samples, labels);
const command_line_parser::option_type& option_alg = parser.option("a");
if (!option_alg) { if (!option_alg) {
// Do KRR if user didn't specify an algorithm // Do KRR if user didn't specify an algorithm
std::cout << "No algorithm specified, default to KRR\n"; std::cout << "No algorithm specified, default to KRR\n";
......
...@@ -21,7 +21,7 @@ exp10_ (double m) ...@@ -21,7 +21,7 @@ exp10_ (double m)
void void
option_range::set_option ( option_range::set_option (
clp& parser, command_line_parser& parser,
std::string const& option, std::string const& option,
float default_val float default_val
) )
......
...@@ -32,7 +32,7 @@ public: ...@@ -32,7 +32,7 @@ public:
max_value = 100; max_value = 100;
incr = 10; incr = 10;
} }
void set_option (clp& parser, std::string const& option, void set_option (command_line_parser& parser, std::string const& option,
float default_val); float default_val);
float get_min_value (); float get_min_value ();
float get_max_value (); float get_max_value ();
......
...@@ -20,7 +20,7 @@ using namespace std; ...@@ -20,7 +20,7 @@ using namespace std;
static const char* static const char*
get_kernel ( get_kernel (
clp& parser command_line_parser& parser
) )
{ {
const char* kernel = "rbk"; const char* kernel = "rbk";
...@@ -34,7 +34,7 @@ get_kernel ( ...@@ -34,7 +34,7 @@ get_kernel (
static void static void
get_rbk_gamma ( get_rbk_gamma (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
option_range& range option_range& range
) { ) {
...@@ -47,7 +47,7 @@ get_rbk_gamma ( ...@@ -47,7 +47,7 @@ get_rbk_gamma (
static void static void
get_krls_tolerance ( get_krls_tolerance (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
option_range& range option_range& range
) )
...@@ -60,7 +60,7 @@ get_krls_tolerance ( ...@@ -60,7 +60,7 @@ get_krls_tolerance (
static double static double
get_mlp_hidden_units ( get_mlp_hidden_units (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples std::vector<dense_sample_type>& dense_samples
) )
{ {
...@@ -75,7 +75,7 @@ get_mlp_hidden_units ( ...@@ -75,7 +75,7 @@ get_mlp_hidden_units (
static double static double
get_mlp_num_iterations ( get_mlp_num_iterations (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples std::vector<dense_sample_type>& dense_samples
) )
{ {
...@@ -90,7 +90,7 @@ get_mlp_num_iterations ( ...@@ -90,7 +90,7 @@ get_mlp_num_iterations (
static void static void
get_svr_c ( get_svr_c (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
option_range& range option_range& range
) )
...@@ -103,7 +103,7 @@ get_svr_c ( ...@@ -103,7 +103,7 @@ get_svr_c (
static double static double
get_svr_epsilon_insensitivity ( get_svr_epsilon_insensitivity (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples std::vector<dense_sample_type>& dense_samples
) )
{ {
...@@ -125,7 +125,7 @@ get_svr_epsilon_insensitivity ( ...@@ -125,7 +125,7 @@ get_svr_epsilon_insensitivity (
void void
krls_test ( krls_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
) )
...@@ -176,7 +176,7 @@ krls_test ( ...@@ -176,7 +176,7 @@ krls_test (
static void static void
krr_rbk_test ( krr_rbk_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
) )
...@@ -229,7 +229,7 @@ krr_rbk_test ( ...@@ -229,7 +229,7 @@ krr_rbk_test (
static void static void
krr_lin_test ( krr_lin_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
) )
...@@ -250,7 +250,7 @@ krr_lin_test ( ...@@ -250,7 +250,7 @@ krr_lin_test (
void void
krr_test ( krr_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
) )
...@@ -271,7 +271,7 @@ krr_test ( ...@@ -271,7 +271,7 @@ krr_test (
void void
mlp_test ( mlp_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
) )
...@@ -321,7 +321,7 @@ mlp_test ( ...@@ -321,7 +321,7 @@ mlp_test (
void void
svr_test ( svr_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
) )
......
...@@ -12,28 +12,28 @@ ...@@ -12,28 +12,28 @@
void void
krr_test ( krr_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
); );
void void
krls_test ( krls_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
); );
void void
mlp_test ( mlp_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
); );
void void
svr_test ( svr_test (
clp& parser, command_line_parser& parser,
std::vector<dense_sample_type>& dense_samples, std::vector<dense_sample_type>& dense_samples,
std::vector<double>& labels std::vector<double>& labels
); );
......
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