kennethduong commited on
Commit
41e8fd4
·
1 Parent(s): dc7b249
expense_forecast/Expense_Forecasting.py CHANGED
@@ -51,7 +51,7 @@ def get_input_data(income, interest_rate, inflation_rate, holidays):
51
 
52
  def forecast_expense(model_fit, input_data, df):
53
  if (input_data['Income (VND)'].iloc[0] < 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
 
 
51
 
52
  def forecast_expense(model_fit, input_data, df):
53
  if (input_data['Income (VND)'].iloc[0] < 5000000):
54
+ return input_data['Income (VND)'].iloc[0] * 0.78492
55
  forecast = model_fit.predict(start=len(df), end=len(df), exog=input_data)
56
  return forecast.iloc[0]
57