Quantcast
Channel: Why is a simple Binary classification failing in a feedforward neural network? - Stack Overflow
Browsing index pages (2 articles)
↧

Why is a simple Binary classification failing in a feedforward neural network?

I am new to Pytorch. I was trying to model a binary classifier on the Kepler dataset. The following was my dataset class.class KeplerDataset(Dataset): def __init__(self, test=False):...

View Article


Answer by guibs35 for Why is a simple Binary classification failing in a...

You are optimizing many times (100 steps) on the first batch (first samples), then moving to the next samples. It means that your model will overfit your few samples before going to the next batch....

View Article

Browsing index pages (2 articles)


Latest Images