This project demonstrates the integration of PyTorch for running a neural network and Raylib for real-time visualization. The neural network is trained using PyTorch, and the trained model's predictions are visualized using the Raylib library.
Make sure you have the following dependencies installed:
- Python 3.x
- PyTorch
- Raylib
- NumPy
Install the required Python packages using:
pip install torch raylib numpy
- Clone the repository:
git clone https://github.com/your-username/your-project.git
cd your-project
- Run the main script:
python main.py
- Observe the Raylib window displaying the neural network's predictions in real-time.
main.py
: The main script that runs the PyTorch neural network and uses Raylib for visualization.model.py
: Contains the PyTorch neural network model definition.data.py
: Handles data loading and preprocessing.visualization.py
: Manages the visualization using Raylib.
- PyTorch: https://pytorch.org/
- Raylib: https://www.raylib.com/
This project is licensed under the MIT License - see the LICENSE file for details.