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.
16 lines
343 B
16 lines
343 B
2 years ago
|
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()
|
||
|
|