kennethduong commited on
Commit
0adf255
·
1 Parent(s): c6b62e0

fix income below 5000000 vnd

Browse files
expense_forecast/Expense_Forecasting.py CHANGED
@@ -50,6 +50,8 @@ def get_input_data(income, interest_rate, inflation_rate, holidays):
50
  return input_data
51
 
52
  def forecast_expense(model_fit, input_data, df):
 
 
53
  forecast = model_fit.predict(start=len(df), end=len(df), exog=input_data)
54
  return forecast.iloc[0]
55
 
 
50
  return input_data
51
 
52
  def forecast_expense(model_fit, input_data, df):
53
+ if (input_data['Income (VND)'] < 5000000):
54
+ return input_data['Income (VND)'] * 0.78492
55
  forecast = model_fit.predict(start=len(df), end=len(df), exog=input_data)
56
  return forecast.iloc[0]
57