azrai99 commited on
Commit
0a98de7
·
verified ·
1 Parent(s): c0f7916

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -156,13 +156,14 @@ def select_model(horizon, model_type, max_steps=200):
156
  raise ValueError(f"Unsupported model type: {model_type}")
157
 
158
  def model_train(df,model):
 
159
  model.fit(df)
160
  return model
161
 
162
  def forecast_time_series(df, model_type, horizon, max_steps=200):
163
  start_time = time.time() # Start timing
164
  freq = determine_frequency(df)
165
- st.sidebar.write(f"Determined frequency: {freq}")
166
 
167
  selected_model = select_model(horizon, model_type, max_steps)
168
  model = model_train(df, selected_model)
 
156
  raise ValueError(f"Unsupported model type: {model_type}")
157
 
158
  def model_train(df,model):
159
+ df['ds'] = pd.to_datetime(df['ds'])
160
  model.fit(df)
161
  return model
162
 
163
  def forecast_time_series(df, model_type, horizon, max_steps=200):
164
  start_time = time.time() # Start timing
165
  freq = determine_frequency(df)
166
+ st.sidebar.write(f"Data frequency: {freq}")
167
 
168
  selected_model = select_model(horizon, model_type, max_steps)
169
  model = model_train(df, selected_model)