Skip to content

The repository includes two classes for constructing a neural network model for tabular data using PyTorch Lightning

Notifications You must be signed in to change notification settings

VascoSch92/Pytorch_NNmodel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 

Repository files navigation

Neural network for tabular data with pytprch lightning

The repository includes two classes for constructing a neural network model for tabular data using PyTorch Lightning. The script NNdataset.py comprises the code for generating a dataset, while the script NNmodel.py encompasses the code for building a neural network model.

Feel free to share any advice or constructive criticism you may have to enhance the code.

About

The repository includes two classes for constructing a neural network model for tabular data using PyTorch Lightning

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages