UniTS pipeline (#581)

* UniTS pipeline

* S-1 result

* S-1 result

* fix map_location

* remove matplotlib

* remove packages

* remove matplotlib

* incorporated comments

* syntax changes

* fix line breaks

* load_path fixed

Author: Linh-nk

orion/pipelines/pretrained/UniTS/UniTS.json

@@ -0,0 +1,22 @@
+{
+    "primitives": [
+        "mlstars.custom.timeseries_preprocessing.time_segments_aggregate",
+        "sklearn.impute.SimpleImputer",
+        "sklearn.preprocessing.StandardScaler",
+        "orion.primitives.units.UniTS",
+        "orion.primitives.timeseries_errors.regression_errors",
+        "orion.primitives.timeseries_anomalies.find_anomalies"
+    ],
+    "init_params": {
+        "mlstars.custom.timeseries_preprocessing.time_segments_aggregate#1": {
+            "time_column": "timestamp",
+            "interval": 21600,
+            "method": "mean"
+        }
+    }, 
+    "output_names": {
+        "orion.primitives.timeseries_anomalies.find_anomalies#1": {
+            "y": "anomalies"
+        }
+    }
+}

orion/primitives/jsons/orion.primitives.units.UniTS.json

@@ -0,0 +1,85 @@
+{
+    "name": "orion.primitives.units.UniTS",
+    "contributors": [
+        "Linh Nguyen <[email protected]>"
+    ],
+    "documentation": "https://arxiv.org/abs/2403.00131",
+    "description": "UniTS for timeseries forecasting",
+    "classifiers": {
+        "type": "estimator",
+        "subtype": "regressor"
+    },
+    "modalities": [],
+    "primitive": "orion.primitives.units.UniTS",
+    "produce": {
+        "method": "predict",
+        "args": [
+            {
+                "name": "X",
+                "type": "ndarray"
+            }, 
+            {
+                "name": "index",
+                "type": "ndarray"
+            }
+        ],
+        "output": [
+            {
+                "name": "y_hat",
+                "type": "ndarray"
+            }, 
+            {
+                "name": "y",
+                "type": "ndarray"
+            },
+            {
+                "name": "index",
+                "type": "ndarray"
+            }
+        ]
+    },
+    "hyperparameters": {
+        "fixed": {
+            "window_size": {
+                "type": "int",
+                "default": 250
+            },
+            "pred_len": {
+                "type": "int",
+                "default": 1
+            },
+            "prompt_num": {
+                "type": "int",
+                "default": 10
+            },
+            "d_model": {
+                "type": "int",
+                "default": 64
+            },
+            "patch_len": {
+                "type": "int",
+                "default": 1
+            },
+            "step": {
+                "type": "int",
+                "default": 1
+            },
+            "dropout": {
+                "type": "float",
+                "default": 0.1
+            },
+            "e_layers": {
+                "type": "int",
+                "default": 3
+            },
+            "n_heads": {
+                "type": "int",
+                "default": 8
+            },
+            "load_path": {
+                "type": "str", 
+                "default": null
+            }
+        }
+    }
+}