Function main

src/thundersvm/thundersvm-train.cpp:19–127

#endif

int main(int argc, char **argv) {
    try {
		el::Loggers::addFlag(el::LoggingFlag::FixedTimeFormat);
        CMDParser parser;
        parser.parse_command_line(argc, argv);
        DataSet train_dataset;
        train_dataset.load_from_file(parser.svmtrain_input_file_name);
        std::shared_ptr<SvmModel> model;
        switch (parser.param_cmd.svm_type) {
            case SvmParam::C_SVC:
                model.reset(new SVC());
                LOG(INFO) << "training C-SVC";
                LOG(INFO) << "C = " << parser.param_cmd.C;
                break;
            case SvmParam::NU_SVC:
                model.reset(new NuSVC());
                LOG(INFO) << "training nu-SVC";
                LOG(INFO) << "nu = " << parser.param_cmd.nu;
                break;
            case SvmParam::ONE_CLASS:
                model.reset(new OneClassSVC());
                LOG(INFO) << "training one-class SVM";
                LOG(INFO) << "C = " << parser.param_cmd.C;
                break;
            case SvmParam::EPSILON_SVR:
                model.reset(new SVR());
                LOG(INFO) << "training epsilon-SVR";
                LOG(INFO) << "C = " << parser.param_cmd.C << " p = " << parser.param_cmd.p;
                break;
            case SvmParam::NU_SVR:
                model.reset(new NuSVR());
                LOG(INFO) << "training nu-SVR";
                LOG(INFO) << "nu = " << parser.param_cmd.nu;
                break;
        }

        //todo add this to check_parameter method
        if (parser.param_cmd.svm_type == SvmParam::NU_SVC) {
            train_dataset.group_classes();
            for (int i = 0; i < train_dataset.n_classes(); ++i) {
                int n1 = train_dataset.count()[i];
                for (int j = i + 1; j < train_dataset.n_classes(); ++j) {
                    int n2 = train_dataset.count()[j];
                    if (parser.param_cmd.nu * (n1 + n2) / 2 > min(n1, n2)) {
                        printf("specified nu is infeasible\n");
                        return 1;
                    }
                }
            }
        }
        if (parser.param_cmd.kernel_type != SvmParam::LINEAR)
            if (!parser.gamma_set) {
                parser.param_cmd.gamma = 1.f / train_dataset.n_features();
                LOG(WARNING) << "using default gamma=" << parser.param_cmd.gamma;
            } else {
                LOG(INFO) << "gamma = " << parser.param_cmd.gamma;
            }