experiments

class cogdl.experiments.AutoML(task, dataset, model, n_trials=3, **kwargs)

Bases: object

Args:

func_search: function to obtain hyper-parameters to search

run()

cogdl.experiments.auto_experiment(task: str, dataset, model, **kwargs)

cogdl.experiments.check_task_dataset_model_match(task, variants)

cogdl.experiments.experiment(task: str, dataset, model, **kwargs)

cogdl.experiments.gen_variants(**items)

cogdl.experiments.output_results(results_dict, tablefmt='github')

cogdl.experiments.raw_experiment(task: str, dataset, model, **kwargs)

cogdl.experiments.set_best_config(args)

cogdl.experiments.train(args)

cogdl.experiments.variant_args_generator(args, variants)

Form variants as group with size of num_workers