9 lines
193 B
Python
9 lines
193 B
Python
from .Model import Model
|
|
from .cnn import CNNPredictor
|
|
from .transformer import Transformer
|
|
|
|
|
|
model_called: dict[str, type[Model]] = {
|
|
'cnn': CNNPredictor,
|
|
'transformer': Transformer
|
|
}
|