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.NDArrayAdapterXgbNDArrayis the XGBoost implementation ofNDArray.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description voidclose()voiddetach()longgetHandle()Returns the native XGBoost Booster pointer.ai.djl.ndarray.types.SparseFormatgetSparseFormat()voidintern(ai.djl.ndarray.NDArray replaced)voidreturnResource(ai.djl.ndarray.NDManager manager)java.nio.ByteBuffertoByteBuffer()-
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, contentEquals, contentEquals, copyTo, cos, cosh, cumSum, cumSum, div, div, divi, divi, dot, eq, eq, erfinv, exp, expandDims, flatten, flip, floor, gammaln, gather, get, getDataType, getDevice, getGradient, getManager, getName, getNDArrayInternal, getShape, getUid, gt, gt, gte, gte, hasGradient, inverse, isInfinite, isNaN, 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, repeat, repeat, repeat, repeat, reshape, rotate90, round, sequenceMask, sequenceMask, set, set, set, set, set, setName, setRequiresGradient, setScalar, sign, signi, sin, sinh, softmax, sort, sort, split, split, sqrt, square, squeeze, 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, equals, finalize, getClass, hashCode, 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, get, get, get, get, get, getBoolean, getByte, getDouble, getFloat, getInt, getLong, 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, sum, swapAxes, take, toArray, toBooleanArray, toByteArray, 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:
getSparseFormatin interfaceai.djl.ndarray.NDArray- Overrides:
getSparseFormatin 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:
closein interfacejava.lang.AutoCloseable- Specified by:
closein interfaceai.djl.ndarray.NDArray- Specified by:
closein interfaceai.djl.ndarray.NDResource- Overrides:
closein classai.djl.ndarray.NDArrayAdapter
-
-