from monad.ui.module import readout_sketch


# declare variables
# path to recommendation model checkpoints
checkpoint_path = "<path/to/downstream/model/checkpoints>"

# path to encoded predictions file
predictions_file = "<path/to/predictions/my_predictions.tsv>"

# get a generator yielding entity id and all scored products
sketch_readout_generator = readout_sketch(
    predictions_file=predictions_file,
    checkpoint_path=checkpoint_path,
)