https://github.com/yahoo/graphkit

A lightweight Python module for creating and running ordered graphs of computations.
https://github.com/yahoo/graphkit

computer-vision machine-learning python

Last synced: about 2 months ago
JSON representation

A lightweight Python module for creating and running ordered graphs of computations.

• Host: GitHub
• URL: https://github.com/yahoo/graphkit
• Owner: yahoo
• License: apache-2.0
• Created: 2016-06-17T23:41:57.000Z (about 8 years ago)
• Default Branch: master
• Last Pushed: 2023-03-21T02:49:20.000Z (over 1 year ago)
• Last Synced: 2024-06-19T18:09:12.702Z (3 months ago)
• Topics: computer-vision, machine-learning, python
• Language: Python
• Homepage:
• Size: 109 KB
• Stars: 82
• Watchers: 13
• Forks: 24
• Open Issues: 21
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-systematic-trading - GraphKit - with-python](https://img.shields.io/badge/Made%20with-Python-1f425f.svg) | (Graph Computation / Cryptocurrencies)
• awesome-systematic-trading - GraphKit(No activity) - A lightweight Python module for creating and running ordered graphs of computations. (Basic Components / Computation)

README

        
# GraphKit

[![PyPI version](https://badge.fury.io/py/graphkit.svg)](https://badge.fury.io/py/graphkit) [![Build Status](https://travis-ci.org/yahoo/graphkit.svg?branch=master)](https://travis-ci.org/yahoo/graphkit) [![codecov](https://codecov.io/gh/yahoo/graphkit/branch/master/graph/badge.svg)](https://codecov.io/gh/yahoo/graphkit)

[Full Documentation](https://pythonhosted.org/graphkit/)

> It's a DAG all the way down

## Lightweight computation graphs for Python

GraphKit is a lightweight Python module for creating and running ordered graphs of computations, where the nodes of the graph correspond to computational operations, and the edges correspond to output --> input dependencies between those operations.  Such graphs are useful in computer vision, machine learning, and many other domains.

## Quick start

Here's how to install:

```

pip install graphkit

```    

Here's a Python script with an example GraphKit computation graph that produces multiple outputs (`a * b`, `a - a * b`, and `abs(a - a * b) ** 3`):

```

from operator import mul, sub

from graphkit import compose, operation

# Computes |a|^p.

def abspow(a, p):

    c = abs(a) ** p

    return c

# Compose the mul, sub, and abspow operations into a computation graph.

graph = compose(name="graph")(

    operation(name="mul1", needs=["a", "b"], provides=["ab"])(mul),

    operation(name="sub1", needs=["a", "ab"], provides=["a_minus_ab"])(sub),

    operation(name="abspow1", needs=["a_minus_ab"], provides=["abs_a_minus_ab_cubed"], params={"p": 3})(abspow)

)

# Run the graph and request all of the outputs.

out = graph({'a': 2, 'b': 5})

# Prints "{'a': 2, 'a_minus_ab': -8, 'b': 5, 'ab': 10, 'abs_a_minus_ab_cubed': 512}".

print(out)

# Run the graph and request a subset of the outputs.

out = graph({'a': 2, 'b': 5}, outputs=["a_minus_ab"])

# Prints "{'a_minus_ab': -8}".

print(out)

```

As you can see, any function can be used as an operation in GraphKit, even ones imported from system modules!

# License

Code licensed under the Apache License, Version 2.0 license. See LICENSE file for terms.