Module gbdt::input

This module implements the data loader.

Currently we support to kind of input format: csv format and libsvm data format.

Example

LibSVM format

use gbdt::input::InputFormat;
use gbdt::input;
let test_file = "data/xgb_binary_logistic/agaricus.txt.test";
let mut fmt = input::InputFormat::txt_format();
fmt.set_feature_size(126);
fmt.set_delimeter(' ');
let test_data = input::load(test_file, fmt);

CSV format

use gbdt::input::InputFormat;
use gbdt::input;
let test_file = "data/xgb_multi_softmax/dermatology.data.test";
let mut fmt = InputFormat::csv_format();
fmt.set_feature_size(34);
let test_data = input::load(test_file, fmt);

Structs

• InputFormat
  The input file format struct.

Enums

• FileFormat
  This enum type defines the data file format.

Functions

• infer
  Function used for input file type inference. This can help recognize the file format. If the file is in csv type, this function also helps to check whether the csv file has header.
• load
  Load file with certain input format.
• load_csv
  Load csv file.
• load_txt
  Load txt file.