Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Write classification model output to a file for later use in reports. #451

Merged
merged 6 commits into from
May 4, 2021

Conversation

Shruthi42
Copy link
Contributor

@Shruthi42 Shruthi42 commented May 4, 2021

A file model_outputs.csv is written with columns subject, prediction_target, label, model_output and cross_validation_split. This file is not written out for sequence models.

This PR also changes the signature for the function generate_custom_report in ModelConfigBase.

melanibe
melanibe previously approved these changes May 4, 2021
ml_util.set_random_seed(config.get_effective_random_seed(), "Model Testing")
ds = config.get_torch_dataset_for_inference(data_split).as_data_loader(
shuffle=False,
batch_size=1,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think the batch size was set to 1 in the past and it is not related to this PR. However, if you think that inference metric computation takes long, we could do the inference, metric computation, and logging in batches as well.

Copy link
Contributor

@ozan-oktay ozan-oktay left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@melanibe melanibe merged commit daefdba into main May 4, 2021
@melanibe melanibe deleted the shbannur/multiclass-report branch May 4, 2021 14:04
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants