An open API service indexing awesome lists of open source software.

https://github.com/mujadded/svm-linear-classifier-with-pytorch


https://github.com/mujadded/svm-linear-classifier-with-pytorch

Last synced: 8 days ago
JSON representation

Awesome Lists containing this project

README

          

# Linear Classifier with SVM loss funtion

A Simple implementation of Linear Classifier with SVM loss funtion using pyTorch.

This uses GPU and also vectorized Python to be most optimized.

## Files

`linear_classifier.py` has the main code for Linear Classifier with SVM

`test_classifier.ipynb` has the playgroud where this is tested.

Example of Linear Classifier:

```
Trainning Accuracy: 0.34, Validation Accuracy: 0.33
Final Accuracy on Test: 33.00%
```