DMatrixBuilder

public class DMatrixBuilder

Collection of utility functions to create data matrices

Author

Hyunsu Cho

Methods

LoadDatasetFromLibSVM

public static List<DataPoint> LoadDatasetFromLibSVM(String filename)

Load a LIBSVM data file and construct a list of data points

Parameters

• filename – path to LIBSVM file

Throws

• TreeliteError – Treelite error

• IOException – IO error

Returns

Created list of data points

createDenseDMatrix

public static DMatrix createDenseDMatrix(Iterator<DataPoint> dIter)

Assemble a dense matrix from a list of data points (float32)

Parameters

• dIter – Iterator of data points

Throws

• TreeliteError – Treelite error

• IOException – IO error

Returns

Created dense batch

createSparseCSRDMatrix

public static DMatrix createSparseCSRDMatrix(Iterator<DataPoint> dIter)

Build a sparse (CSR layout) matrix from a list of data points (float32)

Parameters

• dIter – Iterator of data points

Throws

• TreeliteError – Treelite error

• IOException – IO error

Returns

Created sparse data matrix