Math Solver AI is a neural network designed to solve simple arithmetic operations. It is built using PyTorch and includes functionalities to train, evaluate, and test the model. The model can be trained on a dataset of arithmetic problems, and its performance can be evaluated on a test set. Additionally, users can input their own arithmetic problems to see the model's predictions.
- Train the Model: Train the neural network on a dataset of arithmetic problems.
- Evaluate the Model: Evaluate the model's performance on a test set.
- Test Custom Inputs: Input your own arithmetic problems to see the model's predictions.
- Save and Load Checkpoints: Save the model's progress and load it later to resume training.
- Python 3.6+
- PyTorch 1.6.0+
- pandas
- scikit-learn
- matplotlib
-
Clone the repository:
git clone https://github.com/JimmyVS/Math_Solver_AI.git cd Math_Solver_AI
-
Install the required packages:
pip install torch pandas scikit-learn matplotlib
-
Prepare your dataset as a CSV file with columns
Number A
,Operation
,Number B
, andSolution
. An example dataset is provided below:Number A Operation Number B Solution 1 + 1 2 2 - 1 1 3 * 2 6 4 / 2 2 -
Run the script:
python math_solver.py
-
Follow the on-screen instructions to train, evaluate, or test the model.
This repository is under a MIT license. This repository was made by JimmyVS. Please do not claim as your's.