module ai.dlbase
¶
Short summary¶
module code_beatrix.ai.dlbase
Base class for deep learning models.
Classes¶
class |
truncated documentation |
---|---|
Implements a common interface to manipulate pre-trained deep learning models. |
|
Implements a common interface to manipulate pre-trained deep learning models processing images. |
Methods¶
method |
truncated documentation |
---|---|
Log something. |
|
|
Log something. |
Applies the model on features X. |
|
|
Applies the model on features X. |
Documentation¶
Base class for deep learning models.
- class code_beatrix.ai.dlbase.DeepLearningBase(model, gpu, fLOG=None)¶
Bases :
object
Implements a common interface to manipulate pre-trained deep learning models.
- Paramètres:
model – model (url, filename, …)
gpu – use gpu
fLOG – logging function
- __init__(model, gpu, fLOG=None)¶
- Paramètres:
model – model (url, filename, …)
gpu – use gpu
fLOG – logging function
- log(*args, **kwargs)¶
Log something.
- predict(X)¶
Applies the model on features X.
- Paramètres:
X – features
- Renvoie:
prediction
- class code_beatrix.ai.dlbase.DeepLearningImage(model, gpu=False, fLOG=None)¶
Bases :
DeepLearningBase
Implements a common interface to manipulate pre-trained deep learning models processing images.
- Paramètres:
model – model name
gpu – use gpu
fLOG – logging function
- __init__(model, gpu=False, fLOG=None)¶
- Paramètres:
model – model name
gpu – use gpu
fLOG – logging function