We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 6b081a1 + 2667ed8 commit f35f379Copy full SHA for f35f379
scripts/pykg2vec_train.py
@@ -11,7 +11,7 @@
11
def main():
12
args = KGEArgParser().get_args(sys.argv[1:])
13
14
- knowledge_graph = KnowledgeGraph(dataset=args.dataset_name)
+ knowledge_graph = KnowledgeGraph(dataset=args.dataset_name, custom_dataset_path=args.dataset_path)
15
knowledge_graph.prepare_data()
16
17
config_def, model_def = Importer().import_model_config(args.model_name.lower())
0 commit comments