Package ai.djl.ml.xgboost

Class XgbModel

java.lang.Object
ai.djl.BaseModel
ai.djl.ml.xgboost.XgbModel
All Implemented Interfaces:
ai.djl.Model, AutoCloseable
public class XgbModel extends ai.djl.BaseModel
XgbModel is the XGBoost implementation of Model.

  • Field Summary

    Fields inherited from class ai.djl.BaseModel

    artifacts, block, dataType, inputData, manager, modelDir, modelName, properties, wasLoaded

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    close()
    void
    load(Path modelPath, String prefix, Map<String,?> options)

    Methods inherited from class ai.djl.BaseModel

    describeInput, describeOutput, finalize, getArtifact, getArtifact, getArtifactAsStream, getArtifactNames, getBlock, getDataType, getModelPath, getName, getNDManager, getProperties, getProperty, load, newPredictor, newTrainer, paramPathResolver, readParameters, save, setBlock, setDataType, setModelDir, setProperty, toString

    Methods inherited from class java.lang.Object

    clone, equals, getClass, hashCode, notify, notifyAll, wait, wait, wait

    Methods inherited from interface ai.djl.Model

    cast, getProperty, load, load, load, newPredictor, quantize

  • Method Details