import objective_b
import os

if os.name == 'posix':
    user = 'roye'
else:
    user = 'idan'


def module_4_model_a_creation(model_type, model_a_mimic_cohort_csv, model_a_eicu_cohort_csv):
    objective_b.main(given_model_type=model_type, given_mimic_data_path=model_a_mimic_cohort_csv,
                     given_eicu_data_path=model_a_eicu_cohort_csv)


def main():
    model_type = 'a'
    data_path_mimic = f'C:/tools/external_validation_set_{model_type}_train_data.csv' if user == 'idan' \
        else f'/Users/user/Documents/University/Workshop/external_validation_set_{model_type}_train_data.csv'
    data_path_eicu = 'C:/tools/model_a_eicu_cohort_training_data.csv' if user == 'idan' \
        else '/Users/user/Documents/University/Workshop/model_a_eicu_cohort_training_data.csv'
    module_4_model_a_creation(model_type, data_path_mimic, data_path_eicu)
