search

jlloh / catboost-rs

Unofficial Rust bindings for Catboost
8 stars 3 forks source link

Allow for multiclass classification #11

Open iKintosh opened 4 days ago

iKintosh commented 4 days ago

Hi, I noticed that this binding works only with binary classification. I added small change to allow for multiclass classification. Ideally it should put vec! inside vec!, but it requires changes in a couple of places as well.

jlloh commented 4 days ago

Hello, thank you for your contribution!

  1. Would it be possible to add some unit basic tests for multiclass classifications?
  2. We should probably bump the package version in Cargo.toml ?
  3. Perhaps also worth starting a changelog.md file, I can add one and you can rebase from master and add your changes to the changelog