Package ai.djl.ml.lightgbm

Class LgbmNDManager

java.lang.Object

ai.djl.ndarray.BaseNDManager

ai.djl.ml.lightgbm.LgbmNDManager

All Implemented Interfaces:

ai.djl.ndarray.NDManager, AutoCloseable

public class LgbmNDManager
extends ai.djl.ndarray.BaseNDManager

LgbmNDManager is the LightGBM implementation of NDManager.

Nested Class Summary

Nested classes/interfaces inherited from class ai.djl.ndarray.BaseNDManager

ai.djl.ndarray.BaseNDManager.TempResource

Nested classes/interfaces inherited from interface ai.djl.ndarray.NDManager

ai.djl.ndarray.NDManager.SystemNDManager

Field Summary

Fields inherited from class ai.djl.ndarray.BaseNDManager

alternativeManager, capped, closed, device, name, parent, resources, tempResources, uid

Method Summary

Modifier and Type	Method	Description
ByteBuffer	allocateDirect(int capacity)
ai.djl.ndarray.NDArray	create(Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
LgbmNDArray	from(ai.djl.ndarray.NDArray array)
ai.djl.engine.Engine	getEngine()
ai.djl.ndarray.NDList	load(Path path)
ai.djl.ndarray.NDManager	newSubManager(ai.djl.Device device)

Methods inherited from class ai.djl.ndarray.BaseNDManager

arange, attachInternal, attachUncappedInternal, cap, close, copyBuffer, create, create, createCoo, createCSR, createRowSparse, debugDump, defaultDevice, detachInternal, eye, full, getDevice, getManagedArrays, getName, getParentManager, invoke, invoke, isOpen, linspace, newSubManager, randomInteger, randomMultinomial, randomMultinomial, randomNormal, randomPermutation, randomUniform, sampleGamma, sampleGamma, sampleNormal, sampleNormal, samplePoisson, samplePoisson, setName, tempAttachInternal, toString, truncatedNormal, validateBuffer

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.ndarray.NDManager

arange, arange, arange, arange, arange, arange, arange, arange, attachAll, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, createCSR, createCSR, createRowSparse, decode, decode, eye, eye, eye, eye, full, full, full, hanningWindow, linspace, linspace, linspace, linspace, load, ones, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, ret, tempAttachAll, truncatedNormal, truncatedNormal, truncatedNormal, zeros, zeros, zeros

Method Details

allocateDirect

public ByteBuffer allocateDirect(int capacity)

from

public LgbmNDArray from(ai.djl.ndarray.NDArray array)

newSubManager

public ai.djl.ndarray.NDManager newSubManager(ai.djl.Device device)

getEngine

public ai.djl.engine.Engine getEngine()

create

public ai.djl.ndarray.NDArray create(Buffer data,
 ai.djl.ndarray.types.Shape shape,
 ai.djl.ndarray.types.DataType dataType)

load

public ai.djl.ndarray.NDList load(Path path)

Specified by:

load in interface ai.djl.ndarray.NDManager

Overrides:

load in class ai.djl.ndarray.BaseNDManager