hgupta01 / svm_classification_keras

Implementing SVM layer for classification using Keras Module

svm_classification_keras

This repo contains an example of MNIST Data classification using linear SVM following the paper below using Keras with TF backend. https://arxiv.org/abs/1306.0239

The loss function used is multi-class classification.

I also implemented the example in Tensorflow (using tf.data.Dataset), but training is slow, and better examples are available on GitHub (svm_layer, and cnn_svm).