Update restful/onnx_utilities.py

restful/onnx_utilities.py

@@ -17,7 +17,7 @@ class Utilities:
     #     return float(Decimal(str(val)).quantize(Decimal('0.01'), rounding=ROUND_DOWN))
     def truncate_2_decimal(self, val: float):
         try:
-            return float(Decimal(str(float(val))).quantize(Decimal('0.0001'), rounding=ROUND_DOWN))
+            return float(Decimal(str(float(val))).quantize(Decimal('0.001'), rounding=ROUND_DOWN))
         except Exception as e:
             print("Decimal error:", e)
             return float(val)
@@ -68,7 +68,7 @@ class Utilities:
             # last_date += pd.Timedelta(days=1)
             last_date = pd.to_datetime(last_date) + pd.Timedelta(days=1)
             # predicted_prices[last_date.strftime('%Y-%m-%d')] = float(denorm_price)
-            predicted_prices[last_date] = float(denorm_price)
+            predicted_prices[last_date] = self.truncate_2_decimal(denorm_price)
         
             lst_seq = np.roll(lst_seq, shift=-1, axis=1)
             lst_seq[:, -1, -1] = value
@@ -87,7 +87,7 @@ class Utilities:
         close_denorm = self.denormalization(close_values, min_close, max_close)
 
         actuals = [
-            {'date': to_datetime(date).strftime('%Y-%m-%d'), 'price': float(price)}
+            {'date': to_datetime(date).strftime('%Y-%m-%d'), 'price': self.truncate_2_decimal(price)}
             for date, price in zip(df_date, close_denorm)
         ]