More Mem logo memo: more mem

1 2 3 4 5 6

You can log to a list, but memo also allows you to log to other places.

import numpy as np
from memo import memlist, memfile, memfunc, memweb

data = []

@memweb("https://someurl.com/logging-api/")
@memfunc(callback=print)
@memfile(filepath="results.jsonl")
@memlist(data=data)
def birthday_experiment(class_size, n_sim=1000):
    """Simulates the birthday paradox. Vectorized = Fast!"""
    sims = np.random.randint(1, 365 + 1, (n_sim, class_size))
    sort_sims = np.sort(sims, axis=1)
    n_uniq = (sort_sims[:, 1:] != sort_sims[:, :-1]).sum(axis = 1) + 1
    return {"est_prob": np.mean(n_uniq != class_size), "number": 42}

for size in range(2, 100):
    birthday_experiment(class_size=size, n_sim=1_000)