Update handler.py

handler.py

@@ -113,6 +113,15 @@ class EndpointHandler:
         # Decode the tokens back into samples
         scenarios = self.tokenizer.decode(tokens)[:, -steps:]
         print(f"Generated {n_scenarios} scenarios in {time.time() - t_start:.2f} seconds ⏱")
+        print("Nans:", torch.isnan(scenarios).sum().item())
+        print("Infs:", torch.isinf(scenarios).sum().item())
+        high_not_highest = (scenarios[..., :4].max(-1).values > scenarios[..., 1])
+        low_not_lowest = (scenarios[..., :4].min(-1).values < scenarios[..., 2])
+        invalid_candle = high_not_highest | low_not_lowest
+        print("Highest not high rate:", high_not_highest.float().mean().item())
+        print("Lowest not low rate:", low_not_lowest.float().mean().item())
+        print("Invalid candles rate:", invalid_candle.float().mean().item())
+        print("Invalid scenario rate:", invalid_candle.any(dim=-1).float().mean().item())
         return {
             "timestamps": (timestamps[-1] + torch.arange(1, steps+1) * torch.median(torch.diff(timestamps)).item()).tolist(),
             "open": scenarios[:, :, 0].tolist(),