dotnet / dotnet/machinelearning

Support One Class Classification in Ml.net

Open
#5,878 3 comments 0 reactions 0 assignees View on GitHub
area-Trainers
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.