debug_merlion_random_skater.py

from timemachines.skatertools.data.real import hospital
import random
from timemachines.skaters.mrln.merlioninclusion import using_merlion
from timemachines.skating import prior

if using_merlion:
    from timemachines.skaters.mrln.allmerlionskaters import MERLION_SKATERS

    def test_random_skater(show=False):
        k = 3
        n = 100
        f = random.choice(MERLION_SKATERS)
        print( f.__name__ )
        y = hospital(n=n)
        neg_y = [-yt for yt in y]
        e = [-1]*90+[100]+[-1]*90
        x, x_std = prior(f=f, y=neg_y, k=k, e=e )


if __name__ == '__main__':
    assert using_merlion, 'pip install salesforce-merlion'
    print(MERLION_SKATERS)
    import time
    for _ in range(100):
        st = time.time()
        test_random_skater()
        print(time.time()-st)