Skip to content
This repository has been archived by the owner on Sep 7, 2022. It is now read-only.

Commit

Permalink
Implement run.py to run an experiment given a configuration
Browse files Browse the repository at this point in the history
  • Loading branch information
mali-git committed Jan 10, 2019
1 parent 92e01e1 commit 9a8e2aa
Showing 1 changed file with 30 additions and 0 deletions.
30 changes: 30 additions & 0 deletions src/biokeen/run.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
# -*- coding: utf-8 -*-

"""Script for starting the pipeline and saving the results."""

from dataclasses import dataclass
from typing import Dict, Mapping, Optional

import pykeen

from pykeen.utilities.pipeline import Pipeline


@dataclass
class Results:
"""Results from PyKEEN."""

config: Mapping
pipeline: Pipeline
results: Mapping


def run(config: Dict,
output_directory: Optional[str] = None) -> Results:
"""Run PyKEEN using a given configuration."""
results = pykeen.run(
config=config,
output_directory=output_directory,
)

return results

0 comments on commit 9a8e2aa

Please sign in to comment.