MTG / DCASE-models

Python library for rapid prototyping of environmental sound analysis systems
MIT License
42 stars 5 forks source link

Add more machine learning backends #15

Open pzinemanas opened 3 years ago

pzinemanas commented 3 years ago

pytorch and scikit-learn are good candidates.

elcronos commented 3 years ago

Hi @pzinemanas. I'm a PhD student and like me, many in the academia has a preference for Pytorch. However, in terms of Environmental Sound Classification models, the majority of source code available is implemented on tensorflow/keras. I reckon a pytorch version of this library would be a great contribution !

pzinemanas commented 3 years ago

Hi @elcronos , thank you very much for using the library and giving your valuable feedback. That's true, most of the researchers are now using pytorch. We are working on this (see for instance #24).