You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
task_predictions_test should include the test_fold_str.
_score_epoch_end() should include an optional prediction_pkl_name: Optional[str] = None parameter. If it is not None (because we passed in test_fold_str), the final prediction_logit and target should be dumped to a pickle file.
For event models, we should also dump predicted_events and target_events and maybe prediction.
The idea is that if we do a long expensive prediction on some model, we want the final test output saved. That way we could do a more detailed analysis, e.g. seeing what the scores are for each class.
The text was updated successfully, but these errors were encountered:
task_predictions_test
should include thetest_fold_str
._score_epoch_end() should include an optional
prediction_pkl_name: Optional[str] = None
parameter. If it is not None (because we passed in test_fold_str), the finalprediction_logit
andtarget
should be dumped to a pickle file.For event models, we should also dump
predicted_events
andtarget_events
and maybeprediction
.The idea is that if we do a long expensive prediction on some model, we want the final test output saved. That way we could do a more detailed analysis, e.g. seeing what the scores are for each class.
The text was updated successfully, but these errors were encountered: