Source code for runner_estimation
import joblib
from estimation import ChoiceSetGenerator
from helper_func import parse_df_schedule
DATA = joblib.load('../data/example_data_estimation.joblib')
TT = joblib.load('../data/example_tt.joblib')
PARAMS = '../data/target_params_032023.joblib'
MH_PARAMS = {"n_iter":1000,
"n_burn": 50,
"n_skip": 1,
"uniform": False,
}
N_ALT = 5
[docs]def main():
data = parse_df_schedule(DATA)
estimator = ChoiceSetGenerator(DATA, PARAMS, n_alt = 5, mh_params=MH_PARAMS)
estimator.run()
train_wide, train_long, test = estimator.train_test_sets()
if __name__ == '__main__':
main()