nip.graph_isomorphism

nip.graph_isomorphism#

All components for the graph isomorphism task.

Has classes for:

  • Handling data

  • Defining the RL environment

  • Generating a dataset

  • Building agents

Examples

>>> from nip.parameters import HyperParameters, Scenario, Trainer
>>> from nip.graph_isomorphism import GraphIsomorphismAgentsBuilder
>>> hyper_params = HyperParameters(
...     "graph_isomorphism", "solo_agent", "eru10000"
... )
>>> agents = GraphIsomorphismAgentsBuilder.build(hyper_params, "cpu")

Modules

agents

Graph isomorphism agents components.

data

Dataset classes for the graph isomorphism experiments.

dataset_generation

Functions for generating a dataset of pairs of graphs with WL scores.

environment

The graph isomorphism RL environment.

rollout_samples

Visualiser for graph isomorphism rollout samples.