lib.sedna.backend.torch

Package Contents

Classes

TorchBackend

ML Framework Backend base Class

class lib.sedna.backend.torch.TorchBackend(estimator, fine_tune=True, **kwargs)

Bases: sedna.backend.base.BackendBase

ML Framework Backend base Class

evaluate(**kwargs)

evaluate model.

train(**kwargs)

Not implemented!

predict(data, **kwargs)

Inference model.

load(model_url='', model_name=None, **kwargs)