This is a PyQt5-based desktop application that allows you to visualize the training and prediction of a simple neural network model.
pip install -r requirements.txt
- Clone the repository to your local machine.
- Install the required packages by running
pip install -r requirements.txtin the project directory. - Run the application by executing
python main.pyin the project directory.
The application allows you to specify the input data and model hyperparameters through a series of input widgets. Once the input data and hyperparameters are specified, you can train the model by clicking the "Fit" button, and visualize the predictions by clicking the "Predict" button.
The left panel of the application displays the input widgets, while the right panel displays a plot of the ground truth and predicted values.
This project is licensed under the MIT License. See the LICENSE file for more information.
- Add support for saving and loading trained models.
- Add support for different activation functions.
- Add support for different optimization algorithms.