def save(ctx, cfg): '''Saves predictions''' with ceph.connect(cfg) as c: c.insert_predictions(merlin.functions.denumpify(ctx['predictions'])) return ctx
def delete(ctx, cfg): '''Delete existing predictions''' with ceph.connect(cfg) as c: c.delete_predictions(ctx['cx'], ctx['cy']) return ctx
def load_model(ctx, cfg): with ceph.connect(cfg) as c: ctile = c.select_tile(ctx['tx'], ctx['ty']) model = bytes.fromhex(first(ctile)['model']) if model is None: raise Exception( "No model found for tx:{tx} and ty:{ty}".format(**ctx)) else: return assoc(ctx, 'model_bytes', model)
def save(ctx, cfg): '''Saves an xgboost model for this tx & ty''' # will need to decode hex when pulling model # >>> bytes.fromhex('deadbeef') #b'\xde\xad\xbe\xef' logger.info("saving model") model_bytes = segaux.bytes_from_booster(ctx['model']).hex() ctx['model'] = None del ctx['model'] with ceph.connect(cfg) as c: c.insert_tile(ctx['tx'], ctx['ty'], model_bytes) return ctx
def segments(ctx, cfg): '''Return saved segments''' with ceph.connect(cfg) as c: return assoc(ctx, 'segments', c.select_segments(ctx['cx'], ctx['cy']))
def segments(ctx, cfg): '''Return saved segments''' logger.info("getting segments for cx:{} cy:{}".format( ctx['cx'], ctx['cy'])) with ceph.connect(cfg) as c: return assoc(ctx, 'segments', c.select_segments(ctx['cx'], ctx['cy']))