Line too long (105 > 79 characters):
1 from gds.common.metrics.all_metrics import Accuracy, MultiTaskAccuracy, MSE, multiclass_logits_to_pred, \Line too long (87 > 79 characters):
11 # 'multitask_accuracy': MultiTaskAccuracy(prediction_fn=multiclass_logits_to_pred),Line too long (90 > 79 characters):
12 # 'multitask_binary_accuracy': MultiTaskAccuracy(prediction_fn=binary_logits_to_pred),Line too long (85 > 79 characters):
21 'multitask_accuracy': MultiTaskAccuracy(prediction_fn=multiclass_logits_to_pred),Line too long (88 > 79 characters):
22 'multitask_binary_accuracy': MultiTaskAccuracy(prediction_fn=binary_logits_to_pred),Line too long (94 > 79 characters):
34 models = ['gin', 'gcn', 'gin_virtual', 'gcn_virtual', 'cheb', 'cheb_virtual', '3wlgnn', 'mlp',Line too long (106 > 79 characters):
44 schedulers = ['linear_schedule_with_warmup', 'cosine_schedule_with_warmup', 'ReduceLROnPlateau', 'StepLR',Line too long (115 > 79 characters):
48 losses = ['cross_entropy', 'lm_cross_entropy', 'MSE', 'multitask_bce', 'fasterrcnn_criterion', 'BCEWithLogitsLoss']