Skip to content

gmatasci/PyTorchLinearSVM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

PyTorchLinearSVM

Implementation

Implements a simple Linear SVM (soft margin formulation) trained in PyTorch by gradient descent on a 2D toy dataset with binary labels. The hyper-parameter search (C parameter) is done by cross-validation via scikit-learn. It includes a visualization of the decision function and associated support vectors throughout the epochs of training for various C values. Inspiration from:

Results

hyperplane_C_0p01

hyperplane_C_100

results

About

Simple linear SVM (soft margin formulation) trained in PyTorch by gradient descent.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages