PyOrganoid is the world's first^* Python package for the simulation of organoids for the purpose of studying Organoid Intelligence (OI) and Organoid Learning (OL). It is designed to be simple to use and easy to extend with support for standard machine learning libraries such as TensorFlow, PyTorch, and Scikit-Learn (as well as ONNX-format models).

• Documentation: https://danielathome19.github.io/pyorganoid
• Source code: https://github.com/danielathome19/pyorganoid
• Bug reports: https://github.com/danielathome19/pyorganoid/issues
• Contributing: CONTRIBUTING.md

It provides:

• a simple and intuitive API
• support for standard machine learning libraries
• a growing library of organoid models
• visualization tools for organoid simulation
• numerous simulation environments and scheduling algorithms
• TODO: support for parallel/distributed computing, bio/cheminformatics libraries, logging, and more

PyOrganoid is currently in development and may not yet be ready for production use. We are actively seeking contributors to help us improve the package and expand its capabilities. If you are interested in contributing, please see our contributing guide.

^*As of July 6th, 2024, to the best of our knowledge :)

Installation

Pip

PyOrganoid can be installed (without built-in support for machine learning libraries) using pip:

pip install pyorganoid

To include support for all machine learning libraries, use:

pip install pyorganoid[all]

Or, to include support for a specific library (TensorFlow, PyTorch, Scikit-Learn, or ONNX), use:

pip install pyorganoid[tensorflow]
pip install pyorganoid[torch]
pip install pyorganoid[sklearn]
pip install pyorganoid[onnx]

Conda

PyOrganoid can also be installed using conda:

conda install -c danielathome19 pyorganoid

To include support for extras, use pip as described above.

Quickstart

For a quick introduction to PyOrganoid, see the Spiking Neuron Test in the test directory. This test demonstrates the creation of a simple spiking neuron organoid running a binary classification Multi-Layer Perceptron (MLP) model using TensorFlow.

If you prefer Scikit-Learn, PyTorch, or ONNX models, see the Volumetric Organoid Test (Scikit-Learn), the Gene Regulation Organoid Test (PyTorch), or the Immune Response Organoid Test (ONNX), respectively.

For example:

import pyorganoid as po

model = po.tf_module.TFModel("model.h5")
env = po.environments.Environment()
organoid = po.organoids.SpikingNeuronOrganoid(env, model, num_cells=50)
organoid.plot_organoid(show_properties=True)

scheduler = po.simulation.Scheduler(organoid)
scheduler.simulate(steps=100)

organoid.plot_simulation_history(title="Spiking Neuron Organoid Simulation", 
                                 y_label="Membrane Potential (mV)", dpi=600)

License

PyOrganoid is licensed under the BSD-3 License. See the LICENSE file for more information.