10 lines
269 B
Python
10 lines
269 B
Python
from .Model import Model
|
|
from .autoencoder import AutoEncoder
|
|
from .cnn import CNNPredictor
|
|
from .transformer import ByteTransformer
|
|
|
|
model_called: dict[str, type[Model]] = {
|
|
'cnn': CNNPredictor,
|
|
'transformer': ByteTransformer,
|
|
'autoencoder': AutoEncoder
|
|
}
|