Skip to content
/ sgd Public

Stochastic Gradient Descent for Binary Classification task

1 star 0 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

alexey-sidnev/sgd

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
utils
utils
 
 
README.md
README.md
 
 
train.py
train.py
 
 

Repository files navigation

Stochastic Gradient Descent for Binary Classification task

Supports only Fully-Connected layers and Cross-Entropy loss.

Requirements

  1. Python 3.x
  2. NumPy

Usage

python3 train.py <train.csv> <test.csv> <submission.csv>

TODO

  1. Add regularization.
  2. Batch elements balancing.

About

Stochastic Gradient Descent for Binary Classification task

Resources

Readme
Activity

Stars

1 star

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages