Skip to content

Clustering with nature inspired algorithms for PyTorch

License

Notifications You must be signed in to change notification settings

Catastropha/pyriad

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Clustering with nature inspired algorithms

Codacy Badge Codacy Badge Build Status Version Downloads License

pyriad offers clustering with a variety of nature inspired algorithms built with Python on top of the deep learning library PyTorch.

You can extend pyriad according to your own needs. You can implement custom algorithms by extending simple abstract classes. Pyriad is highly parallelizable and transferable to GPU.

Algorithms

As of today, the following algorithms have been implemented:

  • Particle Swarm Optimization (PSO) [1]
  • Cuckoo Search (CS) [2]
  • Grey Wolf Optimization (GWO) [3]
  • Flower Pollination Algorithm (FP) [4]

Installation

  1. Install PyTorch. You can find it here: PyTorch
  2. pip install pyriad

Examples

You can find examples in examples/ directory

You can also run examples: python examples/pso_iris.py

You might want to export PYTHONPATH=/path/to/this/directory

Contribute

  1. Implement new algorithms
  2. Improve code design
  3. Improve comments and readme
  4. Tests

About

Clustering with nature inspired algorithms for PyTorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages