lib.sedna.backend.torch

Package Contents

Classes

TorchBackend

ML Framework Backend base Class
class lib.sedna.backend.torch.TorchBackend(estimator, fine_tune=True, **kwargs)[source]

Bases: sedna.backend.base.BackendBase

ML Framework Backend base Class

evaluate(self, **kwargs)[source]

evaluate model.

train(self, **kwargs)[source]

Not implemented!

predict(self, data, **kwargs)[source]

Inference model.

load(self, model_url='', model_name=None, **kwargs)[source]