From a50e0f6b2caa390abaeb7c7247a7cd9af49a6e2f Mon Sep 17 00:00:00 2001 From: Eric Potash Date: Thu, 4 Oct 2018 10:13:45 -0500 Subject: [PATCH] fix max year --- lead/model/cv.py | 3 ++- lead/model/workflows.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/lead/model/cv.py b/lead/model/cv.py index ac9908c..73caae8 100644 --- a/lead/model/cv.py +++ b/lead/model/cv.py @@ -7,12 +7,13 @@ from datetime import date import pandas as pd import numpy as np +import os import logging from drain.util import lru_cache YEAR_MIN = 2003 -YEAR_MAX = 2017 +YEAR_MAX = pd.Timestamp(os.environ['TODAY']).year @lru_cache(maxsize=10) def lead_data(month, day, wic_lag): diff --git a/lead/model/workflows.py b/lead/model/workflows.py index 8c5f7ee..0380e9f 100644 --- a/lead/model/workflows.py +++ b/lead/model/workflows.py @@ -93,7 +93,7 @@ def bll6_models(estimators, cv_search={}, transform_search={}, dump_estimator=Fa """ cvd = dict( - year=range(2013, 2014+1), + year=range(2011, 2014+1), month=1, day=1, train_years=[6],