Skip to content

Support One Class Classification in Ml.net #5878

Open
@Marios7

Description

@Marios7

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    imageBugs related image datatype tasks

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions