You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
343 B
15 lines
343 B
import settings |
|
|
|
from link_prediction import Link_pred_Runner |
|
|
|
dataname = 'cora' # 'cora' or 'citeseer' or 'pubmed' |
|
model = 'DBGAN' # 'arga_ae' or 'DBGAN' |
|
task = 'link_prediction' |
|
|
|
settings = settings.get_settings(dataname, model, task) |
|
|
|
if task == 'link_prediction': |
|
runner = Link_pred_Runner(settings) |
|
|
|
runner.erun() |
|
|
|
|