tdpeuter/2025ML-project-neural_compression
tdpeuter/2025ML-project-neural_compression
Archived
1
Fork 0
Code Issues Pull requests Releases 1 Activity Actions

feat: Add model choice

Browse source
This commit is contained in:
Tibo De Peuter 2025-12-06 21:52:31 +01:00
parent bb241154d9
commit ef50d6321e
Signed by: tdpeuter
GPG key ID: 38297DE43F75FFE2
10 changed files with 102 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/models/__init__.py
View file

@ -1,2 +1,9 @@
from .Model import Model
from .cnn import CNNPredictor
from .transformer import Transformer
from .transformer import Transformer
model_called: dict[str, type[Model]] = {
'cnn': CNNPredictor,
'transformer': Transformer
}