/svm-pytorch

Linear SVM with PyTorch

Primary LanguagePythonMIT LicenseMIT

Support Vector Machines

  • Support Vector Machines (SVMs) with Linear Kernel
  • Stochastic Gradient Descent (SGD)

Requirements

$ pip install matplotlib numpy scikit-learn
$ pip install "torch>=0.4.0"

Example

python main.py --c 0.01 --batchsize 1
Hard margin
c = 0, batchsize = 1		 Soft margin
c = 0.01, batchsize = 1