module ai.dlbase

Inheritance diagram of code_beatrix.ai.dlbase

Short summary

module code_beatrix.ai.dlbase

Base class for deep learning models.

source on GitHub

Classes

class truncated documentation
DeepLearningBase Implements a common interface to manipulate pre-trained deep learning models.
DeepLearningImage Implements a common interface to manipulate pre-trained deep learning models processing images.

Methods

method truncated documentation
__init__  
__init__  
log Log something.
log Log something.
predict Applies the model on features X.
predict Applies the model on features X.

Documentation

Base class for deep learning models.

source on GitHub

class code_beatrix.ai.dlbase.DeepLearningBase(model, gpu, fLOG=None)[source]

Bases : object

Implements a common interface to manipulate pre-trained deep learning models.

source on GitHub

Paramètres:
  • model – model (url, filename, …)
  • gpu – use gpu
  • fLOG – logging function

source on GitHub

__init__(model, gpu, fLOG=None)[source]
Paramètres:
  • model – model (url, filename, …)
  • gpu – use gpu
  • fLOG – logging function

source on GitHub

log(*l, **p)[source]

Log something.

source on GitHub

predict(X)[source]

Applies the model on features X.

Paramètres:X – features
Retourne:prediction

source on GitHub

class code_beatrix.ai.dlbase.DeepLearningImage(model, gpu=False, fLOG=None)[source]

Bases : code_beatrix.ai.dlbase.DeepLearningBase

Implements a common interface to manipulate pre-trained deep learning models processing images.

source on GitHub

Paramètres:
  • model – model name
  • gpu – use gpu
  • fLOG – logging function

source on GitHub

__init__(model, gpu=False, fLOG=None)[source]
Paramètres:
  • model – model name
  • gpu – use gpu
  • fLOG – logging function

source on GitHub