linear_model.ZINB_grad

the ZINB-Grad, a gradient-based ZINB GLMM with GPU acceleration, high-performance scalability, and memory-efficient estimation.

Functions

train_ZINB	Trains a ZINB-Grad model.
train_ZINB_with_val	Trains a ZINB-Grad model with validation.
val_ZINB	Returns the validation loss and negative log-likelihood.

Classes

ZINB_Grad

The ZINB-Grad model.