kazuto1011 / svm-pytorch

Linear SVM with PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Linear SVM with PyTorch

License:MIT License


Languages

Language:Python 100.0%