Spaces:
Running
Running
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 |
|