This repository has been archived on 2025-12-23. You can view files and clone it, but you cannot make any changes to it's state, such as pushing and creating new issues, pull requests or comments.
2025ML-project-neural_compr.../src/models/__init__.py
2025-12-10 14:46:10 +01:00

8 lines
200 B
Python

from .Model import Model
from .cnn import CNNPredictor
from .transformer import ByteTransformer
model_called: dict[str, type[Model]] = {
'cnn': CNNPredictor,
'transformer': ByteTransformer
}