igorrivin / svm-pytorch

Linear SVM with PyTorch

Support Vector Machines

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

Dependencies

PyTorch 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