Model Interface
The Model Interface provides functionality for implementing machine learning models.
Overview
The ModelTemplate class provides:
Model architecture definition
Forward pass implementation
Loss function definition
Integration with training framework
Core ModelTemplate Class
File Location
Cython Implementation:
src/AnalysisG/core/model_template.pyxCython Header:
src/AnalysisG/core/model_template.pxd
Methods to Override
- forward()
Model forward pass.
- loss()
Loss function computation.
- predict()
Generate predictions.
See Also
../core/model_template: Core ModelTemplate implementation
../models/overview: Concrete model implementations