Clustering with nature inspired algorithms
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
- Install PyTorch. You can find it here: PyTorch
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
- Implement new algorithms
- Improve code design
- Improve comments and readme
- Tests