Package ai.djl.ml.xgboost
Class XgbNDArray
- java.lang.Object
-
- ai.djl.ndarray.NDArrayAdapter
-
- ai.djl.ml.xgboost.XgbNDArray
-
- All Implemented Interfaces:
ai.djl.ndarray.BytesSupplier
,ai.djl.ndarray.NDArray
,ai.djl.ndarray.NDResource
,java.lang.AutoCloseable
public class XgbNDArray extends ai.djl.ndarray.NDArrayAdapter
XgbNDArray
is the XGBoost implementation ofNDArray
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
close()
void
detach()
long
getHandle()
Returns the native XGBoost Booster pointer.ai.djl.ndarray.types.SparseFormat
getSparseFormat()
void
intern(ai.djl.ndarray.NDArray replaced)
void
returnResource(ai.djl.ndarray.NDManager manager)
java.nio.ByteBuffer
toByteBuffer()
-
Methods inherited from class ai.djl.ndarray.NDArrayAdapter
abs, acos, acosh, add, add, addi, addi, argMax, argMax, argMin, argMin, argSort, asin, asinh, atan, atanh, attach, batchDot, booleanMask, broadcast, cbrt, ceil, clip, complex, contentEquals, contentEquals, copyTo, cos, cosh, cumProd, cumProd, cumSum, cumSum, div, div, divi, divi, dot, eq, eq, equals, erfinv, exp, expandDims, fft, flatten, flatten, flip, floor, gammaln, gather, gatherNd, get, getDataType, getDevice, getGradient, getManager, getName, getNDArrayInternal, getShape, getUid, gt, gt, gte, gte, hasGradient, hashCode, inverse, isInfinite, isNaN, isReleased, log, log10, log2, logicalAnd, logicalNot, logicalOr, logicalXor, logSoftmax, lt, lt, lte, lte, matMul, max, max, maximum, maximum, mean, mean, median, median, min, min, minimum, minimum, mod, mod, modi, modi, mul, mul, muli, muli, neg, negi, neq, neq, nonzero, norm, norm, normalize, oneHot, percentile, percentile, pow, pow, powi, powi, prod, prod, put, real, repeat, repeat, repeat, repeat, reshape, rotate90, round, scatter, sequenceMask, sequenceMask, set, set, set, set, set, setName, setRequiresGradient, setScalar, sign, signi, sin, sinh, softmax, sort, sort, split, split, sqrt, square, squeeze, stft, stopGradient, sub, sub, subi, subi, sum, sum, take, tan, tanh, tempAttach, tile, tile, tile, tile, toDegrees, toDense, toDevice, toRadians, toSparse, toString, toStringArray, toType, trace, transpose, transpose, trunc
-
Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
Methods inherited from interface ai.djl.ndarray.NDArray
all, allClose, allClose, any, argSort, argSort, booleanMask, broadcast, concat, concat, countNonzero, countNonzero, duplicate, encode, fft, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, getResourceNDArrays, getScalar, getUint8, isEmpty, isScalar, isSparse, like, max, mean, min, none, norm, norm, norm, normalize, normalize, oneHot, oneHot, onesLike, prod, reshape, scaleGradient, set, set, set, set, set, shapeEquals, size, size, split, split, squeeze, squeeze, stack, stack, stft, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, toDebugString, toDebugString, toDebugString, toDoubleArray, toFloatArray, toIntArray, toLongArray, toStringArray, toUint8Array, trace, trace, zerosLike
-
-
-
-
Method Detail
-
getHandle
public long getHandle()
Returns the native XGBoost Booster pointer.- Returns:
- the pointer
-
getSparseFormat
public ai.djl.ndarray.types.SparseFormat getSparseFormat()
- Specified by:
getSparseFormat
in interfaceai.djl.ndarray.NDArray
- Overrides:
getSparseFormat
in classai.djl.ndarray.NDArrayAdapter
-
toByteBuffer
public java.nio.ByteBuffer toByteBuffer()
-
intern
public void intern(ai.djl.ndarray.NDArray replaced)
-
returnResource
public void returnResource(ai.djl.ndarray.NDManager manager)
-
detach
public void detach()
-
close
public void close()
- Specified by:
close
in interfacejava.lang.AutoCloseable
- Specified by:
close
in interfaceai.djl.ndarray.NDArray
- Specified by:
close
in interfaceai.djl.ndarray.NDResource
- Overrides:
close
in classai.djl.ndarray.NDArrayAdapter
-
-