Struct gbdt::input::InputFormat

pub struct InputFormat {
    pub ftype: FileFormat,
    pub header: bool,
    pub label_idx: usize,
    pub enable_unknown_value: bool,
    pub delimeter: char,
    pub feature_size: usize,
}

The input file format struct.

Fields

ftype: FileFormat

Data file format

header: bool

Set if ftype is set to FileFormat. Indicates whether the csv has header.

label_idx: usize

Set if ftype is set to FileFormat. Indicates which colume is the data label. (default = 0)

enable_unknown_value: bool

Set if ftype is set to FileFormat. Indicates if we allow unknown value in data file or not.

delimeter: char

Delimeter of the data file.

feature_size: usize

Set if ftype is set to FileFormat. Indicates the total feature size.

Methods

impl InputFormat

pub fn csv_format() -> InputFormat

Return a default CSV input format.

Example

use gbdt::input::InputFormat;
let mut fmt = InputFormat::csv_format();
println!("{}", fmt.to_string());

pub fn txt_format() -> InputFormat

Return a default LibSVM input format.

Example

use gbdt::input::InputFormat;
let mut fmt = InputFormat::txt_format();
println!("{}", fmt.to_string());

pub fn to_string(&self) -> String

Transform the input format to human readable string.

Example

use gbdt::input::InputFormat;
let mut fmt = InputFormat::csv_format();
println!("{}", fmt.to_string());

pub fn set_feature_size(&mut self, size: usize)

Set feature size for the LibSVM input format.

Example

use gbdt::input::InputFormat;
let mut fmt = InputFormat::txt_format();
fmt.set_feature_size(126); // the total feature size

pub fn set_label_index(&mut self, idx: usize)

Set for label index for CSV format.

Example

use gbdt::input::InputFormat;
let mut fmt = InputFormat::csv_format();
fmt.set_label_index(34);

pub fn set_delimeter(&mut self, delim: char)

Set for label index for CSV format.

Example

use gbdt::input::InputFormat;
let mut fmt = InputFormat::txt_format();
fmt.set_delimeter(' ');

Trait Implementations

impl Clone for InputFormat

fn clone(&self) -> InputFormat

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for InputFormat

impl Debug for InputFormat

fn fmt(&self, f: &mut Formatter) -> Result

Formats the value using the given formatter.

impl Serialize for InputFormat

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error> where
    __S: Serializer, 

Serialize this value into the given Serde serializer.

impl<'de> Deserialize<'de> for InputFormat

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
    __D: Deserializer<'de>, 

Deserialize this value from the given Serde deserializer.