src.ml_helper package

Submodules

src.ml_helper.model module

class src.ml_helper.model.Base(**kwargs)

Bases: object

The most base type

metadata = MetaData(bind=None)

src.ml_helper.model.ENGINE_URL = 'postgres://user:password@localhost:5432/ml_helper'

class src.ml_helper.model.Epoch(**kwargs)

Bases: sqlalchemy.ext.declarative.api.Base

created_at

eval_F1

eval_acc

eval_loss

model_id

number

training_F1

training_acc

training_loss

uuid

class src.ml_helper.model.Model(**kwargs)

Bases: sqlalchemy.ext.declarative.api.Base

epoch

id

last_updated

params

uuid

src.ml_helper.model.Session = sessionmaker(class_='Session', bind=Engine(postgres://user:***@localhost:5432/ml_helper), autoflush=True, autocommit=False, expire_on_commit=True)

src.ml_helper.model.engine = Engine(postgres://user:***@localhost:5432/ml_helper)

src.ml_helper.model.init_db()

src.ml_helper.model.metadata = MetaData(bind=None)

src.ml_helper.training module

src.ml_helper.training.delete_model(model_id)

src.ml_helper.training.hash_parameters(params)

src.ml_helper.training.register_epoch_in_db(model_id, epoch_number, **kwargs)

src.ml_helper.training.register_model_in_db(model_id, params)

src.ml_helper.training.retrieve_best_model_params()