jdermody / brightwire

Bright Wire is an open source machine learning library for .NET with GPU support (via CUDA)
https://github.com/jdermody/brightwire/wiki
MIT License
125 stars 19 forks source link

Request: Multi-labels classifier and Multivariate examples. #15

Closed mctheaw closed 6 years ago

mctheaw commented 6 years ago

Hello Jack,

Are there any plan to add multi-labels classifier and multivariate examples?

Brightwire is a great project, keep up the good work.

Thank you

jdermody commented 6 years ago

Hi

Thanks, glad you like BW!

Good suggestion regarding additional classification samples. Did you have a publicly available data set in mind?

mctheaw commented 6 years ago

Thank you for the response :).

Predicting movie Genres from text or image cover would be a good example but I could find just a few white papers without data set.

Some in Mulan's datasets which are real world use cases may suitable.

http://mulan.sourceforge.net/datasets-mlc.html

Or others below.

http://sci2s.ugr.es/keel/multilabel.php https://www.csie.ntu.edu.tw/~cjlin/libsvmtools/datasets/multilabel.html

Thank you :)

jdermody commented 6 years ago

I've added a multi label sample to brightwire/SampleCode/MultiLabel.cs that might help.

The first example trains a single neural network with six output values. The second builds six networks with each outputting a single value. Both give about the same result.

mctheaw commented 6 years ago

Wow that's awesome, thank you.