iiitl / iiitl/Neural-Networks

Build a basic feed-forward network in PyTorch

Open
#8 23 comments 0 reactions 0 assignees View on GitHub
medium track: library
Dominant language
Jupyter Notebook
Stars
1
Forks
11
PR merge metrics
No merged PRs in 30d

Description

Define a simple PyTorch nn.Module with one hidden layer. Write the training loop including the forward pass, Binary Cross Entropy (BCE) loss calculation, and an Adam optimizer step. Train for 50 epochs on quality_binary and report final test metrics.

Contributor guide

Open the contributing guide

Research direction

No file or test is named in the issue; start by locating where quality_binary is defined and where the model experiments live. Use the specified PyTorch training requirements as the scope, and consider the work complete when 50 epochs have run and final test metrics are reported.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
55/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.