Giters

alexey-sidnev / sgd

Stochastic Gradient Descent for Binary Classification task

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

Languages

Language:Python 100.0%