qywok commited on
Commit
99def00
·
verified ·
1 Parent(s): ca1bfd8

Update restful/onnx_utilities.py

Browse files
Files changed (1) hide show
  1. restful/onnx_utilities.py +5 -2
restful/onnx_utilities.py CHANGED
@@ -12,6 +12,9 @@ class Utilities:
12
  self.posttrained_path = './indonesia_stocks/modeling_datas'
13
  self.scaler_path = './indonesia_stocks/min_max'
14
 
 
 
 
15
  def denormalization(self, data, min_value, max_value):
16
  return (data * (max_value - min_value)) + min_value
17
 
@@ -65,7 +68,7 @@ class Utilities:
65
  # for date, price in predicted_prices.items()
66
  # ]
67
  predictions = [
68
- {'date': date, 'price': float(price)}
69
  for date, price in predicted_prices.items()
70
  ]
71
 
@@ -74,7 +77,7 @@ class Utilities:
74
  close_denorm = self.denormalization(close_values, min_close, max_close)
75
 
76
  actuals = [
77
- {'date': to_datetime(date).strftime('%Y-%m-%d'), 'price': float(price)}
78
  for date, price in zip(df_date, close_denorm)
79
  ]
80
 
 
12
  self.posttrained_path = './indonesia_stocks/modeling_datas'
13
  self.scaler_path = './indonesia_stocks/min_max'
14
 
15
+ def truncate_2_decimal(self, val: float):
16
+ return float(Decimal(str(val)).quantize(Decimal('0.01'), rounding=ROUND_DOWN))
17
+
18
  def denormalization(self, data, min_value, max_value):
19
  return (data * (max_value - min_value)) + min_value
20
 
 
68
  # for date, price in predicted_prices.items()
69
  # ]
70
  predictions = [
71
+ {'date': date, 'price': truncate_2_decimal(price)}
72
  for date, price in predicted_prices.items()
73
  ]
74
 
 
77
  close_denorm = self.denormalization(close_values, min_close, max_close)
78
 
79
  actuals = [
80
+ {'date': to_datetime(date).strftime('%Y-%m-%d'), 'price': truncate_2_decimal(price)}
81
  for date, price in zip(df_date, close_denorm)
82
  ]
83