Commit 79e4f4c8 authored by Christian Marius Lillelund's avatar Christian Marius Lillelund
Browse files

improved a print statement, fixed some lining

parent 1ef99e99
......@@ -69,7 +69,7 @@ def main():
print(f"Interim data dictionary: {cfg.INTERIM_DATA_DIR}")
print(f"Processed data dictionary: {cfg.PROCESSED_DATA_DIR}\n")
print(f"Now making 4 XGBoost models based on {dataset_version} ...")
print(f"Now making 4 XGBoost models based on version: {dataset_version} ...")
make_xgb_models.main()
print(f"Completed making models. Models and SHAP plots can be found at:\n" +
......
......@@ -223,7 +223,7 @@ def prepare_data(data: dict, case: str) -> pd.DataFrame:
new_data_df['NumberAts'] = len(new_data_df['Ats'][0].split(","))
df = split_categorical_columns(new_data_df, col='Ats', tag='Ats',
resolution=ATS_RESOLUTION)
resolution=ATS_RESOLUTION)
cols_ats = [str(i)+'Ats' for i in range(1, ATS_RESOLUTION+1)]
header_list = ['Gender', 'BirthYear', 'Cluster',
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment