Santiago Roman commited on
Commit
495d6b8
1 Parent(s): 0b6fd25
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,10 +35,11 @@ data["month"] = pd.to_datetime(data["month"])
35
  data = data.sort_values(by=["cohort_first_product", "cohort_first_month", "month"])
36
 
37
 
38
- def cohort_predict(start_date, cohort_start, product):
39
 
40
  new_seq = generate_new_data(data, cohort_start, start_date, product, model, X_columns_to_drop, 12)
41
  hist_seq = get_sequence(data, cohort_start, product)
 
42
 
43
  fig = plot_example_from_case(hist_seq, new_seq, 25, product)
44
  fig.canvas.draw()
 
35
  data = data.sort_values(by=["cohort_first_product", "cohort_first_month", "month"])
36
 
37
 
38
+ def cohort_predict(cohort_start, start_date, product):
39
 
40
  new_seq = generate_new_data(data, cohort_start, start_date, product, model, X_columns_to_drop, 12)
41
  hist_seq = get_sequence(data, cohort_start, product)
42
+ print(hist_seq.head())
43
 
44
  fig = plot_example_from_case(hist_seq, new_seq, 25, product)
45
  fig.canvas.draw()