JDACS4C-IMPROVE / IMPROVE

Libraries and scripts for basic IMPROVE functionalities
MIT License
1 stars 3 forks source link

Add input_dir to templates #177

Closed jonesse3 closed 6 minutes ago

jonesse3 commented 22 hours ago

Following changes in the following function, add the necessary input_dir to train and infer template scripts:

store_predictions_df(y_pred: np.array, y_col_name: str, stage: str, output_dir: str, input_dir: Optional[str] = None, y_true: Optional[np.array] = None, round_decimals: int = 4) -> None: