Upload modeling_moment.py

modeling_moment.py

@@ -503,6 +503,21 @@ class MomentEmbeddingModel(MomentPreTrainedModel):
             input_mask = torch.ones_like(time_series_values[:, 0, :])
 
         return self.embed(x_enc=time_series_values, input_mask=input_mask, **kwargs)
+    
+    def calculate_n_patches(self, seq_len: int) -> int:
+        """
+        時系列の長さ（seq_len）を与えて、モデルのself.patch_lenとself.strideを使ってn_patchesを計算して返します。
+        strideがNoneの場合はpatch_lenを使用します。
+
+        Args:
+            seq_len (int): 時系列の長さ
+
+        Returns:
+            int: 計算されたn_patchesの数
+        """
+        stride = self.stride if self.stride is not None else self.patch_len
+        n_patches = (seq_len - self.patch_len) // stride + 1
+        return n_patches
 
 
 # refers: https://github.com/moment-timeseries-foundation-model/moment/blob/088b253a1138ac7e48a7efc9bf902336c9eec8d9/momentfm/models/moment.py#L601