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

Commit

Permalink
Make random seed optional
Browse files Browse the repository at this point in the history
  • Loading branch information
mali-git committed Nov 19, 2018
1 parent caa48dc commit d3478ce
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions src/biokeen/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from pykeen.cli import (
_configure_evaluation_specific_parameters, device_prompt, execution_mode_specific_prompt, model_selection_prompt,
output_direc_prompt, training_file_prompt,
)
random_seed_prompt)
from pykeen.constants import EXECUTION_MODE, HPO_MODE, TRAINING_MODE, TRAINING_SET_PATH
from pykeen.predict import start_predictions_piepline
from pykeen.utilities.cli_utils.cli_print_msg_helper import (
Expand Down Expand Up @@ -53,7 +53,7 @@ def prompt_config(connection, rebuild):
config[TRAINING_SET_PATH] = install_bio2bel_module(name=database_name, connection=connection, rebuild=rebuild)
else:
print_training_set_message()
config[TRAINING_SET_PATH] = training_file_prompt(config)
config = training_file_prompt(config)

print_section_divider()

Expand All @@ -78,6 +78,10 @@ def prompt_config(connection, rebuild):

print_section_divider()

# Step 6: Please select a random seed
config = random_seed_prompt(config=config)
print_section_divider()

# Step 7: Query device to train on
config = device_prompt(config=config)
print_section_divider()
Expand Down

0 comments on commit d3478ce

Please sign in to comment.