dotnet / dotnet/machinelearning
Support One Class Classification in Ml.net
- Dominant language
- C#
- Stars
- 9.4k
- Forks
- 2k
- Avg merge
- 2d 20h
- Merged PRs (30d)
- 11
Description
**Is your feature request related to a problem? Please describe.**
I am building an image classifieres using transfer learning, it's all ok till we get the problem of binary classification, in our case it's (Documents vs Not Documents), it does not take long till someone realizes that unlike the data in documents class, the amount of data in the Not Document class is infinite!
**Describe the solution you'd like**
One way to solve this problem is by using a feature extractor tensorflow model(InceptionV3, ResNet ...etc.) - which is possibile in Ml.Net by the way - in order to obtain the features' vector from the penultimate layer of the network, and then feed this vector to another One-Class-Classification/Anomaly Detection algorithm like _One-Class-SVM_ which is NOT available in Ml.net.
**Describe alternatives you've considered**
I tried data augmentation, adding data to the data set, changing the architecture of the network, extracting features and use algorithms other than CNN to train a model with the extracted features' vector, changing the hyperparameters, but non of these methods seems promising.
**Additional context**
We are searching to solve an anomaly detection problem then doing a classification. the way we intend to that in is extracting the features from images with (ResNet50, ResNet101 etc.) to feed them to an anomaly detection algorithm.
Contributor guide
Assessment
This issue has not been assessed yet.