Skip to content

xozero/nn-multiply

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

nn-multiply

This repository contains code for training neural networks to predict the multiplication of two numbers.

To run the experiments yourself, use main.py:

usage: main.py [-h] [--gendata] [--train] [-c CFG_ID] [-n [NRUNS]]

optional arguments:
  -h, --help            show this help message and exit
  --gendata
  --train
  -c CFG_ID, --cfg_id CFG_ID
                        cfg_id
  -n [NRUNS], --nruns [NRUNS]

To configure experiments, edit simulations.json. The CFG_ID in the above command usage is the key in the config. To find all keys in the config, run python3 config.py.

Example commands for generating data for configuration 7_high2_2000 and for training a model afterwards:

python3 main.py --gendata -c 7_high2_2000
python3 main.py --train -c 7_high2_2000

The generated data for all experiments configured in the checked in simulations.json can be found in datasets, it is all checked in. Thus, the --gendata step can be omitted for those.

A blog post with details can be found here.

About

Multiplying large numbers with Neural Networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages