main_visualize_safran.py 808 bytes
from experiment.safran_study.safran import Safran
from itertools import product

from experiment.safran_study.safran_visualizer import SafranVisualizer


def load_all_safran(only_first_one=False):
    all_safran = []
    for safran_alti, nb_day in product([1800, 2400], [1, 3, 7]):
        print('alti: {}, nb_day: {}'.format(safran_alti, nb_day))
        all_safran.append(Safran(safran_alti, nb_day))
        if only_first_one:
            break
    return all_safran


if __name__ == '__main__':
    for safran in load_all_safran(only_first_one=True):
        safran_visualizer = SafranVisualizer(safran)
        # safran_visualizer.visualize_independent_margin_fits(threshold=[None, 20, 40, 60][1])
        # safran_visualizer.visualize_smooth_margin_fit()
        safran_visualizer.visualize_full_fit()