[CLI] initial refactor of CLI

This commit is contained in:
tqchen
2016-01-05 20:43:13 -08:00
parent 0d95e863c9
commit cee148ed64
5 changed files with 357 additions and 340 deletions

View File

@@ -36,6 +36,13 @@ typedef void *BoosterHandle;
*/
XGB_DLL const char *XGBGetLastError();
/*!
* \brief Entry point of CLI program.
* \param argc The number of arguments.
* \param argv The command line arguments.
*/
XGB_DLL int XGBoostCLIMain(int argc, char* argv[])
/*!
* \brief load a data matrix
* \param fname the name of the file

View File

@@ -9,6 +9,7 @@
#include <dmlc/base.h>
#include <dmlc/data.h>
#include <string>
#include <memory>
#include <vector>
#include "./base.h"
@@ -252,7 +253,7 @@ class DMatrix {
* \param fname The file name to be saved.
* \return The created DMatrix.
*/
virtual void SaveToLocalFile(const char* fname);
virtual void SaveToLocalFile(const std::string& fname);
/*!
* \brief Load DMatrix from URI.
* \param uri The URI of input.
@@ -260,7 +261,7 @@ class DMatrix {
* \param load_row_split Flag to read in part of rows, divided among the workers in distributed mode.
* \return The created DMatrix.
*/
static DMatrix* Load(const char* uri,
static DMatrix* Load(const std::string& uri,
bool silent,
bool load_row_split);
/*!