mindsdb / mindsdb_native

Machine Learning in one line of code
http://mindsdb.com
GNU General Public License v3.0
37 stars 28 forks source link

Output class distribution flag #388

Closed paxcema closed 3 years ago

paxcema commented 3 years ago

Why

Having the belief distribution of a predictor is needed in certain use cases (see L#354)

How

Added output_class_distribution flag and corresponding test. Note that the feature is not yet supported for categorical tag columns, due to the way the Lightwood encoder operates.

(Note: the new test will pass once the L#354 PR is merged)