def get_lossid(): db, table = connections.make_db_connection('losses') s = select([table.c.loss_id]) result = db.execute(s).fetchall() if not result: logging.info('Loss ID Retrieval Failed') raise Exception('Loss ID Retrieval Failed') db.dispose() return result[-1]
def get_photos(id=None, limit=None): db, table = connections.make_db_connection('photos') s = select([table]).limit(limit) if id is None else select([table]).where( table.c.photo_id == int(id)) result = db.execute(s).fetchall() if not result: logging.info('Photos Retrieval Failed') raise Exception('Photo Retrieval Failed') db.dispose() return result
def get_photoid(): db, table = connections.make_db_connection('photos') s = select([table.c.photo_id]) result = db.execute(s).fetchall() print((result[-1])[0]) if not result: logging.info('Photo ID Retrieval Failed') raise Exception('Photo ID Retrieval Failed') db.dispose() return (result[-1])[0]
def insert_photos(photo_fname, ranking): db, table = connections.make_db_connection('photos') model_id = selecting.get_modelid() result = db.execute(table.insert().values(photo_fname=str(photo_fname), ranking=int(ranking), model_id=int(model_id))) if not result: logging.info('Photo Insert Failed') raise Exception('Photo Insertion Failed') db.dispose() return selecting.get_modelid()
def insert_loss(modelid, epoch, train_loss, val_loss): db, table = connections.make_db_connection('losses') data = { 'model_id': modelid, 'epoch': epoch, 'train_loss': train_loss, 'validation_loss': val_loss } result = db.execute(table.insert().values(data)) if not result: logging.info('Loss Insert Failed') raise Exception('Loss Insertion Failed') db.dispose() return selecting.get_lossid()
def insert_model(model_name, epochs, start_lr, m_type, num_outputs, loss_fn, filepath): db, table = connections.make_db_connection('models') result = db.execute(table.insert().values(model_name=str(model_name), epochs=int(epochs), start_lr=float(start_lr), m_type=str(m_type), num_outputs=int(num_outputs), loss_fn=str(loss_fn), filepath=str(filepath))) if not result: logging.info('Model Insert Failed') raise Exception('Model Insertion Failed') db.dispose() return selecting.get_modelid()