Skip to content

A demonstration of how to use PyTorch to implement Support Vector Machine with L2 regularizition and multiclass hinge loss

Notifications You must be signed in to change notification settings

HaotianMXu/Multiclass_LinearSVM_with_SGD

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 

Repository files navigation

Multiclass_LinearSVM_with_SGD

A demonstration of how to use PyTorch to implement Support Vector Machine with one-vs.-all hinge loss. Weighted penalty of each class and square hinge loss are also available.

Requirements

  • PyTorch==0.2.0 with GPU support
  • Python==3.5

Approach

  • The key idea is to optimize a linear classifier with one-vs-all Hinge loss proposed by Dr. Weston and Dr. Watkins.
  • For more details, please refer the loss function in the code.

About

A demonstration of how to use PyTorch to implement Support Vector Machine with L2 regularizition and multiclass hinge loss

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages