Kaggle competition: M5 Forecasting - Accuracy
<SOS> metric: M5_WRMSSE Kaggle data source: m5-forecasting-accuracy<define_variables>
warnings.filterwarnings('ignore' )
dls = get_data(fold=0 )<compute_train_metric>
def seed_everything(seed=0):
random.seed(seed)
 np.random.seed(seed)
 def df_parallelize_run(func, t_split):
num_cores = np.min([N_CORES,len(t_split)])
 pool = Pool(num_cores)
 df = pd.concat(pool.map(func, t_split), axis=1)
 pool.close()
pool.join()
return df
def comp_metric(preds, targs, labels=range(len(LABEL_COLS))):
targs = np.eye(4)[targs]
return np.mean([roc_auc_score(targs[:,i], preds[:,i])for i in labels])
 def healthy_roc_auc(*args):
return comp_metric(*args, labels=[0])
 def multiple_diseases_roc_auc(*args):
return comp_metric(*args, labels=[1])
 def rust_roc_auc(*args):
return comp_metric(*args, labels=[2])
 def scab_roc_auc(*args):
return comp_metric(*args, labels=[3] )<choose_model_class>
def get_data_by_store(store):
df = pd.concat([pd.read_pickle(BASE),
pd.read_pickle(PRICE ).iloc[:,2:],
pd.read_pickle(CALENDAR ).iloc[:,2:]],
axis=1)
 df = df[df['store_id']==store]
df2 = pd.read_pickle(MEAN_ENC)[mean_features]
df2 = df2[df2.index.isin(df.index)]
df3 = pd.read_pickle(LAGS ).iloc[:,3:]
df3 = df3[df3.index.isin(df.index)]
df = pd.concat([df, df2], axis=1)
 del df2
df = pd.concat([df, df3], axis=1)
 del df3
features = [col for col in list(df)if col not in remove_features]
df = df[['id','d',TARGET]+features]
df = df[df['d']>=START_TRAIN].reset_index(drop=True)
 return df, features
def get_base_test() :
base_test = pd.DataFrame()
for store_id in STORES_IDS:
temp_df = pd.read_pickle('test_'+store_id+'.pkl')
 temp_df['store_id'] = store_id
base_test = pd.concat([base_test, temp_df] ).reset_index(drop=True)
 return base_test
def make_lag(LAG_DAY):
lag_df = base_test[['id','d',TARGET]]
col_name = 'sales_lag_'+str(LAG_DAY)
 lag_df[col_name] = lag_df.groupby(['id'])[TARGET].transform(lambda x: x.shift(LAG_DAY)).astype(np.float16)
 return lag_df[[col_name]]
def make_lag_roll(LAG_DAY):
shift_day = LAG_DAY[0]
roll_wind = LAG_DAY[1]
lag_df = base_test[['id','d',TARGET]]
col_name = 'rolling_mean_tmp_'+str(shift_day)+'_'+str(roll_wind)
 lag_df[col_name] = lag_df.groupby(['id'])[TARGET].transform(lambda x: x.shift(shift_day ).rolling(roll_wind ).mean())
 return lag_df[[col_name]]
def get_learner(fold_num, lr=1e-3):
opt_func = partial(Adam, lr=lr, wd=0.01, eps=1e-8)
 data = get_data(fold_num)
 model = EfficientNet.from_pretrained("efficientnet-b7", advprop=True)
 model._fc = nn.Linear(2560, data.c)
 learn = Learner(
 dls, model, opt_func=opt_func,
loss_func=LabelSmoothingCrossEntropy() ,
metrics=[
AccumMetric(healthy_roc_auc, flatten=False),
AccumMetric(multiple_diseases_roc_auc, flatten=False),
AccumMetric(rust_roc_auc, flatten=False),
AccumMetric(scab_roc_auc, flatten=False),
AccumMetric(comp_metric, flatten=False)]
).to_fp16()
return learn<find_best_params>
lgb_params = {
'boosting_type': 'gbdt',
'objective': 'tweedie',
'tweedie_variance_power': 1.1,
'metric': 'rmse',
'subsample': 0.5,
'subsample_freq': 1,
'learning_rate': 0.03,
'num_leaves': 2**11-1,
'min_data_in_leaf': 2**12-1,
'feature_fraction': 0.5,
'max_bin': 100,
'n_estimators': 1400,
'boost_from_average': False,
'verbose': -1,
}
get_learner(fold_num=0 ).lr_find()<compute_test_metric>
VER = 1
SEED = 42
seed_everything(SEED)
 lgb_params['seed'] = SEED
N_CORES = psutil.cpu_count()
TARGET = 'sales'
START_TRAIN = 0
END_TRAIN = 1913
P_HORIZON = 28
USE_AUX = True
remove_features = ['id','state_id','store_id',
'date','wm_yr_wk','d',TARGET]
mean_features = ['enc_cat_id_mean','enc_cat_id_std',
'enc_dept_id_mean','enc_dept_id_std',
'enc_item_id_mean','enc_item_id_std']
ORIGINAL = '.. /input/m5-forecasting-accuracy/'
BASE = '.. /input/m5-simple-fe/grid_part_1.pkl'
PRICE = '.. /input/m5-simple-fe/grid_part_2.pkl'
CALENDAR = '.. /input/m5-simple-fe/grid_part_3.pkl'
LAGS = '.. /input/m5-lags-features/lags_df_28.pkl'
MEAN_ENC = '.. /input/m5-custom-features/mean_encoding_df.pkl'
AUX_MODELS = '.. /input/m5-aux-models/'
STORES_IDS = pd.read_csv(ORIGINAL+'sales_train_validation.csv')['store_id']
STORES_IDS = list(STORES_IDS.unique())
 SHIFT_DAY = 28
N_LAGS = 15
LAGS_SPLIT = [col for col in range(SHIFT_DAY,SHIFT_DAY+N_LAGS)]
ROLS_SPLIT = []
for i in [1,7,14]:
for j in [7,14,30,60]:
ROLS_SPLIT.append([i,j] )
def print_metrics(val_preds, val_labels):
comp_metric_fold = comp_metric(val_preds, val_labels)
 print(f'Comp metric: {comp_metric_fold}')
 healthy_roc_auc_metric = healthy_roc_auc(val_preds, val_labels)
 print(f'Healthy metric: {healthy_roc_auc_metric}')
 multiple_diseases_roc_auc_metric = multiple_diseases_roc_auc(val_preds, val_labels)
 print(f'Multi disease: {multiple_diseases_roc_auc_metric}')
 rust_roc_auc_metric = rust_roc_auc(val_preds, val_labels)
 print(f'Rust metric: {rust_roc_auc_metric}')
 scab_roc_auc_metric = scab_roc_auc(val_preds, val_labels)
 print(f'Scab metric: {scab_roc_auc_metric}' )<train_model>
for store_id in STORES_IDS:
print('Train', store_id)
 grid_df, features_columns = get_data_by_store(store_id)
 train_mask = grid_df['d']<=END_TRAIN
valid_mask = train_mask&(grid_df['d']>(END_TRAIN-P_HORIZON))
preds_mask = grid_df['d']>(END_TRAIN-100)
 train_data = lgb.Dataset(grid_df[train_mask][features_columns],
label=grid_df[train_mask][TARGET])
 train_data.save_binary('train_data.bin')
 train_data = lgb.Dataset('train_data.bin')
 valid_data = lgb.Dataset(grid_df[valid_mask][features_columns],
label=grid_df[valid_mask][TARGET])
 grid_df = grid_df[preds_mask].reset_index(drop=True)
 keep_cols = [col for col in list(grid_df)if '_tmp_' not in col]
grid_df = grid_df[keep_cols]
grid_df.to_pickle('test_'+store_id+'.pkl')
 del grid_df
seed_everything(SEED)
 estimator = lgb.train(lgb_params,
train_data,
valid_sets = [valid_data],
verbose_eval = 100, 
)
 model_name = 'lgb_model_'+store_id+'_v'+str(VER)+'.bin'
pickle.dump(estimator, open(model_name, 'wb'))
!rm train_data.bin
del train_data, valid_data, estimator
gc.collect()
MODEL_FEATURES = features_columns
all_val_preds = []
all_val_labels = []
all_test_preds = []
for i in range(N_FOLDS):
print(f'Fold {i} results')
 learn = get_learner(fold_num=i)
 learn.fit_one_cycle(1)
 learn.unfreeze()
learn.fit_one_cycle(1, slice(1e-5, 1e-4))
learn.recorder.plot_loss()
learn.save(f'model_fold_{i}')
 learn.freeze()
learn.data = get_data_larger(i)
 lr_min,_ = learn.lr_find()
learn.fit_one_cycle(1, slice(lr_min))
val_preds, val_labels = learn.get_preds()
print_metrics(val_preds, val_labels)
 all_val_preds.append(val_preds)
 all_val_labels.append(val_labels)
 test_dl = dls.test_dl(test_df)
 test_preds, _ = learn.get_preds(dl=test_dl)
 all_test_preds.append(test_preds)
 plt.show()<compute_test_metric>
all_preds = pd.DataFrame()
base_test = get_base_test()
main_time = time.time()
for PREDICT_DAY in range(1,29):
print('Predict | Day:', PREDICT_DAY)
 start_time = time.time()
grid_df = base_test.copy()
grid_df = pd.concat([grid_df, df_parallelize_run(make_lag_roll, ROLS_SPLIT)], axis=1)
 for store_id in STORES_IDS:
model_path = 'lgb_model_'+store_id+'_v'+str(VER)+'.bin'
if USE_AUX:
model_path = AUX_MODELS + model_path
estimator = pickle.load(open(model_path, 'rb'))
day_mask = base_test['d']==(END_TRAIN+PREDICT_DAY)
 store_mask = base_test['store_id']==store_id
mask =(day_mask)&(store_mask)
 base_test[TARGET][mask] = estimator.predict(grid_df[mask][MODEL_FEATURES])
 temp_df = base_test[day_mask][['id',TARGET]]
temp_df.columns = ['id','F'+str(PREDICT_DAY)]
if 'id' in list(all_preds):
all_preds = all_preds.merge(temp_df, on=['id'], how='left')
 else:
all_preds = temp_df.copy()
print('
' %0.2f min total |' %(( time.time() - main_time)/ 60),
' %0.2f day sales |' %(temp_df['F'+str(PREDICT_DAY)].sum()))
del temp_df
all_preds = all_preds.reset_index(drop=True)
 all_preds
<choose_model_class><EOS> 
submission = pd.read_csv(ORIGINAL+'sample_submission.csv')[['id']]
submission = submission.merge(all_preds, on=['id'], how='left' ).fillna(0)
 submission.to_csv('submission_v'+str(VER)+'.csv', index=False )