Skip to content

friedforfun/PSO-Neural-Network

Repository files navigation

Particle Swarm Optimisation to search for weights in a neural network

In this repository we construct a fully connected artificial neural network from scratch and train it to approximate a pool of functions using particle swarm optimisation.

Requirements

The ANN & PSO library has the following dependencies:

  • Python 3.8
  • Numpy 1.19.1
  • Matplotlib 3.3.2
  • Pandas 1.1.3
  • Scikit-learn 0.23.2
  • tqdm 4.51.0
  • ipython 7.18.1
  • notebook 6.1.1
  • sphinx 3.2.1

Installation

To create a windows virtual environment using conda and the provided spec-file.txt, from the repository root do:

`$ conda create -n <envName> --file spec-file.txt python=3.8`

About

PSO to train a custom neural network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published