lib.sedna.backend.base

Module Contents

Classes

BackendBase

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

ML Framework Backend base Class

property model_name(self)[source]
static parse_kwargs(func, **kwargs)[source]
train(self, *args, **kwargs)[source]

Train model.

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

Inference model.

predict_proba(self, *args, **kwargs)[source]

Compute probabilities of possible outcomes for samples in X.

evaluate(self, *args, **kwargs)[source]

evaluate model.

save(self, model_url='', model_name=None)[source]
model_info(self, model, relpath=None, result=None)[source]
load(self, model_url='', model_name=None, **kwargs)[source]
abstract set_weights(self, weights)[source]

Set weight with memory tensor.

abstract get_weights(self)[source]

Get the weights.