Skip to content
This repository was archived by the owner on Nov 23, 2022. It is now read-only.

Commit fa1b1b6

Browse files
committed
Add missing metric MAD to transformer model
1 parent 7d7adde commit fa1b1b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mise/ml/mlp_mul_transformer_mccr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -515,7 +515,7 @@ def __init__(self, **kwargs):
515515
self.features_nonperiodic = kwargs.get('features_nonperiodic',
516516
["temp", "wind_spd", "wind_cdir", "wind_sdir",
517517
"pres", "humid", "prep"])
518-
self.metrics = kwargs.get('metrics', ['MAE', 'MSE', 'R2'])
518+
self.metrics = kwargs.get('metrics', ['MAE', 'MSE', 'R2', 'MAD'])
519519
self.num_workers = kwargs.get('num_workers', 1)
520520
self.output_dir = kwargs.get(
521521
'output_dir', Path('/mnt/data/MLPTransformerMultivariate/'))

0 commit comments

Comments
 (0)