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)

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)
Paramètres:
  • model – model (url, filename, …)

  • gpu – use gpu

  • fLOG – logging function

source on GitHub

log(*args, **kwargs)

Log something.

source on GitHub

predict(X)

Applies the model on features X.

Paramètres:

X – features

Renvoie:

prediction

source on GitHub

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.

source on GitHub

Paramètres:
  • model – model name

  • gpu – use gpu

  • fLOG – logging function

source on GitHub

__init__(model, gpu=False, fLOG=None)
Paramètres:
  • model – model name

  • gpu – use gpu

  • fLOG – logging function

source on GitHub