scglue.models.base.Model
- class scglue.models.base.Model(*args, **kwargs)[source]
Bases:
object
Abstract model class
- Parameters:
net – Network type
*args – Positional arguments are passed to the network constructor
**kwargs – Keyword arguments are passed to the network constructor
Note
Subclasses may override arguments for API definition.
Methods
Prepare model for training
Alias of
.trainer.fit
.Alias of
.trainer.get_losses
.Save model to file
Upgrade the model if generated by older versions
Attributes
logger
net
Neural network module in the model (read-only)
trainer
Trainer of the neural network module (read-only)