NVE / avalanche_ml

Applying machine learning in the Norwegian Avalanche warning Service
MIT License
2 stars 1 forks source link

Weighting of labels #7

Closed kmunve closed 4 years ago

kmunve commented 4 years ago

Some ML-models allow to weight the labels, by providing a dict with weights. E.g see class_weight for RandomForestClassifier.

With the current setup the same weight-dict is applied to all labels making the code eventually crash. Ideally it should be possible to define a model for each label individually.