⬅ configs/supported.py source

  • E501 Line too long (105 > 79 characters)
1 from gds.common.metrics.all_metrics import Accuracy, MultiTaskAccuracy, MSE, multiclass_logits_to_pred, \
2 binary_logits_to_pred, MultiTaskAveragePrecision
3  
  • F401 'experiments.configs.model.model_defaults' imported but unused
4 from experiments.configs.model import model_defaults
5 from experiments.configs.algorithm import algorithm_defaults
6  
7  
8 # algo_log_metrics = {
9 # 'accuracy': Accuracy(prediction_fn=multiclass_logits_to_pred),
10 # 'mse': MSE(),
  • E501 Line too long (87 > 79 characters)
11 # 'multitask_accuracy': MultiTaskAccuracy(prediction_fn=multiclass_logits_to_pred),
  • E501 Line too long (90 > 79 characters)
12 # 'multitask_binary_accuracy': MultiTaskAccuracy(prediction_fn=binary_logits_to_pred),
13 # 'multitask_avgprec': MultiTaskAveragePrecision(prediction_fn=None),
14 # None: None,
15 # }
16  
17 algo_log_metrics = {
18 'binary_accuracy': Accuracy(prediction_fn=binary_logits_to_pred),
19 'multiclass_accuracy': Accuracy(prediction_fn=multiclass_logits_to_pred),
20 'mse': MSE(),
  • E501 Line too long (85 > 79 characters)
21 'multitask_accuracy': MultiTaskAccuracy(prediction_fn=multiclass_logits_to_pred),
  • E501 Line too long (88 > 79 characters)
22 'multitask_binary_accuracy': MultiTaskAccuracy(prediction_fn=binary_logits_to_pred),
23 'multitask_avgprec': MultiTaskAveragePrecision(prediction_fn=None),
24 None: None,
25 }
26  
27 process_outputs_functions = {
28 'binary_logits_to_pred': binary_logits_to_pred,
29 'multiclass_logits_to_pred': multiclass_logits_to_pred,
30 None: None,
31 }
32  
33 # See models/initializer.py
  • E501 Line too long (94 > 79 characters)
34 models = ['gin', 'gcn', 'gin_virtual', 'gcn_virtual', 'cheb', 'cheb_virtual', '3wlgnn', 'mlp',
35 'gin_15_layers', 'gin_10_layers']
36  
37 # See algorithms/initializer.py
38 algorithms = list(algorithm_defaults.keys())
39  
40 # See optimizer.py
41 optimizers = ['SGD', 'Adam', 'AdamW']
42  
43 # See scheduler.py
  • E501 Line too long (106 > 79 characters)
44 schedulers = ['linear_schedule_with_warmup', 'cosine_schedule_with_warmup', 'ReduceLROnPlateau', 'StepLR',
45 'MultiStepLR']
46  
47 # See losses.py
  • E501 Line too long (115 > 79 characters)
48 losses = ['cross_entropy', 'lm_cross_entropy', 'MSE', 'multitask_bce', 'fasterrcnn_criterion', 'BCEWithLogitsLoss']