Skip to content

Add probabilities associated with the classification predictions

Compare
Choose a tag to compare
@umitbuyuksahin umitbuyuksahin released this 14 Jan 12:52
· 44 commits to main since this release
435d820

Summary

This release provides predicted probabilities associated with the outcomes of
classification problem types. Furthermore, this release makes batch_size, which
is used in batch predictions, configurable.

Features

  • #47: Named the return column as predictions
  • #37: Made the batch_size configurable in PredictionUDF
  • #36: Added probability score associated with the prediction