{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import numpy as np\nimport pandas as pd\nfrom itertools import groupby\nfrom sklearn.model_selection import train_test_split\nfrom pandas.api.types import is_datetime64_ns_dtype\nimport numba\n\nfrom sklearn.ensemble import RandomForestClassifier\nimport xgboost as xgb\nfrom sklearn.ensemble import GradientBoostingClassifier\nfrom sklearn.preprocessing import MinMaxScaler\nimport lightgbm as lgb\nfrom imblearn.under_sampling import RandomUnderSampler\nfrom joblib import Parallel, delayed\nimport gc\nimport plotly.express as px\nfrom metric import score \n\nimport warnings\nwarnings.filterwarnings(\"ignore\")","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:39.313442Z","iopub.execute_input":"2023-10-28T18:16:39.313844Z","iopub.status.idle":"2023-10-28T18:16:39.322762Z","shell.execute_reply.started":"2023-10-28T18:16:39.313813Z","shell.execute_reply":"2023-10-28T18:16:39.321416Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"column_names = {\n    'series_id_column_name': 'series_id',\n    'time_column_name': 'step',\n    'event_column_name': 'event',\n    'score_column_name': 'score',\n}\n\ntolerances = {\n    'onset': [12, 36, 60, 90, 120, 150, 180, 240, 300, 360], \n    'wakeup': [12, 36, 60, 90, 120, 150, 180, 240, 300, 360]\n}","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:39.325006Z","iopub.execute_input":"2023-10-28T18:16:39.325455Z","iopub.status.idle":"2023-10-28T18:16:39.336123Z","shell.execute_reply.started":"2023-10-28T18:16:39.325414Z","shell.execute_reply":"2023-10-28T18:16:39.334934Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"def reduce_mem_usage(df):\n    \n    \"\"\" \n    Iterate through all numeric columns of a dataframe and modify the data type\n    to reduce memory usage.        \n    \"\"\"\n    \n    for col in df.columns:\n        col_type = df[col].dtype\n\n        if col_type != object and not is_datetime64_ns_dtype(df[col]) and not 'category':\n            c_min = df[col].min()\n            c_max = df[col].max()\n            if str(col_type)[:3] == 'int':\n                if c_min > np.iinfo(np.int8).min and c_max < np.iinfo(np.int8).max:\n                    df[col] = df[col].astype(np.int8)\n                elif c_min > np.iinfo(np.int16).min and c_max < np.iinfo(np.int16).max:\n                    df[col] = df[col].astype(np.int16)\n                elif c_min > np.iinfo(np.int32).min and c_max < np.iinfo(np.int32).max:\n                    df[col] = df[col].astype(np.int32)\n                elif c_min > np.iinfo(np.int64).min and c_max < np.iinfo(np.int64).max:\n                    df[col] = df[col].astype(np.int32)  \n            else:\n                df[col] = df[col].astype(np.float16)\n        \n    return df","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:39.338142Z","iopub.execute_input":"2023-10-28T18:16:39.338509Z","iopub.status.idle":"2023-10-28T18:16:39.351518Z","shell.execute_reply.started":"2023-10-28T18:16:39.338479Z","shell.execute_reply":"2023-10-28T18:16:39.350300Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Feature Engineering\n\n**Explanation for initial features ideas**: [notebook](https://www.kaggle.com/code/renatoreggiani/zzzs-feat-eng-ideas-60-memory-reduction) ","metadata":{}},{"cell_type":"code","source":"def get_macd(s, hfast=4, hslow=48, macd=15, center=False):\n    \n    sma_fast = s.rolling(hfast*12*60, min_periods=1, center=center).agg('mean')\n    sma_slow = s.rolling(hslow*12*60, min_periods=1, center=center).agg('mean')\n    macd = (sma_fast - sma_slow).rolling(12*macd, min_periods=1, center=center).mean().astype(np.float32)\n    \n    return macd\n\n\ndef cls_zone(df, col='anglezdiffabs'):\n    \n    s = df[f'{col}']\n    df['anglezdiffabs_macd'] = get_macd(s)\n\n    s = df[f'{col}'].sort_index(ascending=False)\n    df['anglezdiffabs_macd_rev'] = get_macd(s).sort_index()\n    \n    df['macd_spred'] = df['anglezdiffabs_macd']-df['anglezdiffabs_macd_rev']\n    df['macd_spread_diff'] =df['macd_spred'].diff(60)\n    \n#     df.loc[df['macd_spred']<3, 'wakeup_zone'] = 1\n#     df['wakeup_zone'].fillna(0, inplace=True)\n#     df.loc[df['macd_spred']>-3, 'onset_zone'] = 1\n#     df['onset_zone'].fillna(0, inplace=True)\n    \n    return df\n\n\ndef feat_eng(df):\n    \n    df['series_id'] = df['series_id'].astype('category')\n    df['timestamp'] = pd.to_datetime(df['timestamp']).apply(lambda t: t.tz_localize(None))\n    df['hour'] = df[\"timestamp\"].dt.hour\n    df['month'] = df[\"timestamp\"].dt.month\n    \n    df.sort_values(['timestamp'], inplace=True)\n    df.set_index('timestamp', inplace=True)\n    \n    # limit outiliers in enmo\n    df['enmo'] = df['enmo'].clip(upper=4)\n    \n    df['lids'] = np.maximum(0., df['enmo'] - 0.02)\n    df['lids'] = df['lids'].rolling(f'{120*5}s', center=True, min_periods=1).agg('sum')\n    df['lids'] = 100 / (df['lids'] + 1)\n    df['lids'] = df['lids'].rolling(f'{360*5}s', center=True, min_periods=1).agg('mean').astype(np.float32)\n    \n    df['lids_sma_fast'] = df['lids'].rolling('2h', center=True, min_periods=1).agg('mean').astype(np.float32)\n    df['lids_sma_macd'] = (df['lids'].rolling('18h', center=True, min_periods=1).agg('mean') - df['lids_sma_fast']\n                                 ).rolling(f'1h', center=True, min_periods=1).agg('mean').astype(np.float32)\n    \n    df[\"enmo\"] = (df[\"enmo\"]*1000).astype(np.int16)\n    df[\"anglez\"] = df[\"anglez\"].astype(np.int16)\n    df[\"anglezdiffabs\"] = df[\"anglez\"].diff().abs().astype(np.float32)\n    \n    df = cls_zone(df, col='anglezdiffabs')\n\n    for col in ['enmo', 'anglez', 'anglezdiffabs']:\n        \n        df[f'{col}_sma_fast'] = df[f'{col}'].rolling(f'4h', center=True, min_periods=1).agg('mean').astype(np.float32)\n        df[f'{col}_sf_macd'] = (df[f'{col}_sma_fast'] - df[f'{col}'].rolling(f'48h', center=True, min_periods=1).agg('mean')\n                                   ).rolling(f'1h', center=True, min_periods=1).agg('mean').astype(np.float32)\n        \n        # periods in seconds        \n        periods = [60, 300, 720, 1440, 3600] \n        \n        for n in periods:\n            \n            rol_args = {'window':f'{n+5}s', 'min_periods':10, 'center':True}\n            \n            for agg in ['median', 'mean', 'max', 'min', 'std']:\n                df[f'{col}_{agg}_{n}'] = df[col].rolling(**rol_args).agg(agg).astype(np.float32).values\n                gc.collect()\n            \n            if n == max(periods):\n                df[f'{col}_mad_{n}'] = (df[col] - df[f'{col}_median_{n}']).abs().rolling(**rol_args).median().astype(np.float32)\n            \n            df[f'{col}_amplit_{n}'] = df[f'{col}_max_{n}']-df[f'{col}_min_{n}']\n            df[f'{col}_amplit_{n}_min'] = df[f'{col}_amplit_{n}'].rolling(**rol_args).min().astype(np.float32).values\n            \n            df[f'{col}_diff_{n}_max'] = df[f'{col}_max_{n}'].diff().abs().rolling(**rol_args).max().astype(np.float32)\n            df[f'{col}_diff_{n}_mean'] = df[f'{col}_max_{n}'].diff().abs().rolling(**rol_args).mean().astype(np.float32)\n\n    \n            gc.collect()\n        \n#     df.drop(columns=[\n#             'anglezdiffabs_min_720', 'anglezdiffabs_min_3600',\n#             'anglezdiffabs_amplit_720', 'anglezdiffabs_min_60', 'anglezdiffabs_diff_60_max',\n#             'anglezdiffabs','enmo_diff_60_max',\n#             'anglez_median_60', 'anglezdiffabs_amplit_3600',\n#             'enmo_max_60', 'enmo_amplit_720',\n#             'anglezdiffabs_diff_60_mean', 'anglezdiffabs_max_720', 'enmo_median_60',\n#             'anglez_amplit_720', 'enmo_mean_60', \n#             'anglez_mean_720', 'anglez_mean_60',\n#             'anglez_var_720', 'anglez_diff_60_max', 'anglez_diff_720_max',\n#             'enmo_min_60', 'enmo_max_720', 'enmo_min_720', 'anglez_diff_720_mean',\n#             'anglez_diff_60_mean', 'anglez_max_720', 'anglezdiffabs_var_720',\n#             'anglezdiffabs_diff_720_mean', 'enmo_diff_720_mean', 'anglezdiffabs_min_1440', 'anglezdiffabs_min_300',\n#             'anglezdiffabs_amplit_1440', 'anglezdiffabs_amplit_300',\n#             'anglezdiffabs_diff_300_mean', 'enmo_diff_300_mean',\n#             'anglezdiffabs_max_300', 'anglezdiffabs_diff_300_max',\n#             'anglez_diff_300_max', 'enmo_diff_300_max', 'anglez_amplit_300',\n#             'anglez_diff_300_mean', 'enmo_amplit_300', 'anglezdiffabs_var_300',\n#             'enmo_diff_60_mean', 'anglez_mean_300', 'anglez_max_60', 'enmo_var_300',\n#             'anglez_median_300', 'enmo_median_300', 'enmo_amplit_1440',\n#             'enmo_median_720', 'anglez_amplit_1440', 'anglez_mean_1440',\n#             'enmo_max_1440', 'enmo_max_300', 'anglez_max_300', 'enmo_min_1440',\n#             'anglez_var_300', 'enmo_min_3600'\n#                     ], inplace=True)\n    \n    rol_args = {'window':f'8h', 'min_periods':15, 'center':True}\n    df['not_wear'] = df[f'anglez_std_720'].rolling(**rol_args).max().astype(np.float32)\n    \n    \n    df.reset_index(inplace=True)\n    df.dropna(inplace=True)\n    \n#     df = df[(df['wakeup_zone']==1)|(df['onset_zone']==1)]\n    \n\n    return df","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:39.353524Z","iopub.execute_input":"2023-10-28T18:16:39.354032Z","iopub.status.idle":"2023-10-28T18:16:39.391326Z","shell.execute_reply.started":"2023-10-28T18:16:39.353990Z","shell.execute_reply":"2023-10-28T18:16:39.390383Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"file = '/kaggle/input/zzzs-lightweight-training-dataset-target/Zzzs_train.parquet'\n\ndef feat_eng_by_id(idx):\n    \n    from warnings import simplefilter \n    simplefilter(action=\"ignore\", category=pd.errors.PerformanceWarning)\n    \n    df  = pd.read_parquet(file, filters=[('series_id','=',idx)])\n    df['awake'] = df['awake'].astype(np.int8)\n    df = feat_eng(df)\n    \n    return df","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:39.394265Z","iopub.execute_input":"2023-10-28T18:16:39.395023Z","iopub.status.idle":"2023-10-28T18:16:39.407487Z","shell.execute_reply.started":"2023-10-28T18:16:39.394981Z","shell.execute_reply":"2023-10-28T18:16:39.406433Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"DEV = False\n\nseries_id  = pd.read_parquet(file, columns=['series_id'])\nseries_id = series_id.series_id.unique()\n\nprint('nunique id', len(series_id))\n\nif DEV:\n    series_id = series_id[::10]\n    print('DEV nunique ids:', len(series_id))","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:39.409165Z","iopub.execute_input":"2023-10-28T18:16:39.409913Z","iopub.status.idle":"2023-10-28T18:16:41.602614Z","shell.execute_reply.started":"2023-10-28T18:16:39.409849Z","shell.execute_reply":"2023-10-28T18:16:41.601438Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"weird_series = ['31011ade7c0a', 'a596ad0b82aa']\n\nseries_id = [s for s in series_id if s not in weird_series]","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:41.604322Z","iopub.execute_input":"2023-10-28T18:16:41.604763Z","iopub.status.idle":"2023-10-28T18:16:41.610781Z","shell.execute_reply.started":"2023-10-28T18:16:41.604721Z","shell.execute_reply":"2023-10-28T18:16:41.609627Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"%%time\n\ntrain = Parallel(n_jobs=-1)(delayed(feat_eng_by_id)(i) for i in series_id)\ntrain = pd.concat(train, ignore_index=True)\n","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:16:41.612736Z","iopub.execute_input":"2023-10-28T18:16:41.613624Z","iopub.status.idle":"2023-10-28T18:17:18.006531Z","shell.execute_reply.started":"2023-10-28T18:16:41.613583Z","shell.execute_reply":"2023-10-28T18:17:18.004996Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train.shape","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:18.008175Z","iopub.execute_input":"2023-10-28T18:17:18.008604Z","iopub.status.idle":"2023-10-28T18:17:18.017836Z","shell.execute_reply.started":"2023-10-28T18:17:18.008563Z","shell.execute_reply":"2023-10-28T18:17:18.016746Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train = train.loc[train['not_wear']>22]\ntrain.shape","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:18.022183Z","iopub.execute_input":"2023-10-28T18:17:18.023436Z","iopub.status.idle":"2023-10-28T18:17:19.806109Z","shell.execute_reply.started":"2023-10-28T18:17:18.023395Z","shell.execute_reply":"2023-10-28T18:17:19.804688Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"drop_cols = ['series_id', 'step', 'timestamp']\n\nX, y = train.drop(columns=drop_cols+['awake']), train['awake']\n# reduce train dataset by half\nstep = 60\nX, y = X.iloc[::step], y[::step]\nstep","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:19.807723Z","iopub.execute_input":"2023-10-28T18:17:19.808200Z","iopub.status.idle":"2023-10-28T18:17:20.611281Z","shell.execute_reply.started":"2023-10-28T18:17:19.808156Z","shell.execute_reply":"2023-10-28T18:17:20.609033Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"y.value_counts()","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:20.612984Z","iopub.execute_input":"2023-10-28T18:17:20.613647Z","iopub.status.idle":"2023-10-28T18:17:20.626492Z","shell.execute_reply.started":"2023-10-28T18:17:20.613611Z","shell.execute_reply":"2023-10-28T18:17:20.624793Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"del train\ngc.collect()","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:20.628301Z","iopub.execute_input":"2023-10-28T18:17:20.628803Z","iopub.status.idle":"2023-10-28T18:17:20.812615Z","shell.execute_reply.started":"2023-10-28T18:17:20.628758Z","shell.execute_reply":"2023-10-28T18:17:20.811702Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Ensemble Model","metadata":{}},{"cell_type":"code","source":"class EnsembleAvgProba():\n    \n    def __init__(self, classifiers):\n        \n        self.classifiers = classifiers\n    \n    def fit(self,X,y):\n        \n        for classifier in self.classifiers:                \n            classifier.fit(X, y)\n            gc.collect()\n     \n    def predict_proba(self, X):\n        \n        probs = []\n        \n        for m in self.classifiers:\n            probs.append(m.predict_proba(X))\n        \n        probabilities = np.stack(probs)\n        p = np.mean(probabilities, axis=0)\n        \n        return p\n    \n    def predict(self, X):\n        \n        probs = []\n        \n        for m in self.classifiers:\n            probs.append(m.predict(X))\n        \n        probabilities = np.stack(probs)\n        p = np.median(probabilities, axis=0)\n        \n        return p.round()","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:20.814462Z","iopub.execute_input":"2023-10-28T18:17:20.814807Z","iopub.status.idle":"2023-10-28T18:17:20.826651Z","shell.execute_reply.started":"2023-10-28T18:17:20.814779Z","shell.execute_reply":"2023-10-28T18:17:20.825228Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"X.head()","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:20.828277Z","iopub.execute_input":"2023-10-28T18:17:20.828669Z","iopub.status.idle":"2023-10-28T18:17:20.873338Z","shell.execute_reply.started":"2023-10-28T18:17:20.828626Z","shell.execute_reply":"2023-10-28T18:17:20.872173Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"lgb_params1 = {    \n    'boosting_type':'gbdt',\n    'num_leaves':31,\n    'max_depth':6,\n    'learning_rate':0.03,\n    'n_estimators':500,\n    'subsample_for_bin':200000,\n    'min_child_weight':0.001,\n    'min_child_samples':20,\n    'subsample':0.6,\n#     'colsample_bytree':0.7,\n    'reg_alpha':0.025,\n    'reg_lambda':0.025,\n             }\n\nmodel = EnsembleAvgProba(classifiers=[\n                    lgb.LGBMClassifier(random_state=42, **lgb_params1),\n                    GradientBoostingClassifier(n_estimators=100,max_depth=5,min_samples_leaf=300,random_state=42),\n                    RandomForestClassifier(n_estimators=500, min_samples_leaf=300, random_state=42, n_jobs=-1),\n                    xgb.XGBClassifier(n_estimators=520,objective=\"binary:logistic\", learning_rate=0.02, max_depth=7, random_state=42)    \n])","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:20.875142Z","iopub.execute_input":"2023-10-28T18:17:20.876022Z","iopub.status.idle":"2023-10-28T18:17:20.884417Z","shell.execute_reply.started":"2023-10-28T18:17:20.875984Z","shell.execute_reply":"2023-10-28T18:17:20.883234Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"%%time\n\nmodel.fit(X, y)","metadata":{"execution":{"iopub.status.busy":"2023-10-28T18:17:20.885929Z","iopub.execute_input":"2023-10-28T18:17:20.886559Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Features importance\n\nWe can vizualize all fetures in this [notebook](https://www.kaggle.com/code/renatoreggiani/zzzs-interactive-plot-all-feat-ideas), is helpfull to get more insights","metadata":{}},{"cell_type":"code","source":"feats = []\n\nfor m in model.classifiers:\n    feat_imp = m.feature_importances_\n    \n    feat_imp = MinMaxScaler().fit_transform(feat_imp.reshape(-1, 1))\n    feat_imp = pd.Series(pd.Series(feat_imp.reshape(1, -1)[0], index=X.columns).sort_values(), index=X.columns).sort_values()\n    feats.append(feat_imp)","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"feat_imp = pd.Series(pd.concat(feats, axis=1).mean(axis=1), index=feats[0].index).sort_values()\nprint('Columns with poor contribution', feat_imp[feat_imp<0.005].index, sep='\\n')\nfig = px.bar(x=feat_imp, y=feat_imp.index, orientation='h')\nfig.show()","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"feat_imp.sort_values().head(10)","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"# del X, y\ngc.collect()","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Function to get events","metadata":{}},{"cell_type":"code","source":"# def get_events(idx, classifier, file='test_series.parquet') :\n    \n#     test  = pd.read_parquet(f'/kaggle/input/child-mind-institute-detect-sleep-states/{file}',\n#                     filters=[('series_id','=',idx)])\n    \n#     test = feat_eng(test)\n#     test = test[test[f'anglez_std_max_4h']>21]\n    \n# #     test_daily = test.groupby(test['timestamp'].dt.date)[['anglez', 'enmo']].std()\n# #     min_daily_enmo = test_daily['enmo']<5\n# #     days_to_drop = test_daily[min_daily_enmo].index\n# #     test = test[~test['timestamp'].dt.date.isin(days_to_drop)]\n        \n# #     test = test[test['enmo_max_360']>2]\n# #     test = test[test['anglez_std_360']>2]\n    \n#     X_test = test.drop(columns=drop_cols)\n#     test = test[drop_cols]\n    \n#     if test.shape[0]>0:\n\n#         preds, probs = classifier.predict(X_test), classifier.predict_proba(X_test)[:, 1]\n\n#         test['prediction'] = preds\n#         test['prediction'] = test['prediction'].rolling(120+1, center=True).median()\n#         test['probability'] = probs\n\n# #         test = test[test['prediction']!=2]\n\n#         test.loc[test['prediction']==0, 'probability'] = 1-test.loc[test['prediction']==0, 'probability']\n#         test['score'] = test['probability'].rolling(60*12*5, center=True, min_periods=10).mean().bfill().ffill()\n#         test['pred_diff'] = test['prediction'].diff()\n#         test['event'] = test['pred_diff'].replace({1:'wakeup', -1:'onset', 0:np.nan})\n\n#         test_wakeup = test[test['event']=='wakeup'].groupby(test['timestamp'].dt.date).agg('first')\n#         test_onset = test[test['event']=='onset'].groupby(test['timestamp'].dt.date).agg('last')\n#         test = pd.concat([test_wakeup, test_onset], ignore_index=True).sort_values('timestamp')\n#         if test.shape[0]>0:\n#             if test['event'].values[0]=='wakeup':\n#                 test = test[1:]\n\n#     return test\n\n# get_events(series_id[0], model)","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"def get_events(idx, classifier, file='test_series.parquet') :\n    \n    test  = pd.read_parquet(f'/kaggle/input/child-mind-institute-detect-sleep-states/{file}',\n                    filters=[('series_id','=',idx)])\n    \n    test = feat_eng(test)\n    \n    X_test = test.drop(columns=drop_cols)\n    \n    if test.shape[0]>0:\n\n        preds, probs = classifier.predict(X_test), classifier.predict_proba(X_test)[:, 1]\n\n        test['prediction'] = preds\n        test['prediction'] = test['prediction'].rolling(60+1, center=True).median()\n        test['probability'] = probs\n\n        test = test[test['prediction']!=2]\n\n        test.loc[test['prediction']==0, 'probability'] = 1-test.loc[test['prediction']==0, 'probability']\n        test['score'] = test['probability'].rolling(60*12*5, center=True, min_periods=10).mean().bfill().ffill()\n        test['pred_diff'] = test['prediction'].diff()\n        \n        test['event'] = test['pred_diff'].replace({1:'wakeup', -1:'onset', 0:np.nan})\n        test.dropna(subset='event', inplace=True)\n        test['step_diff'] = test['step'].diff()\n        test = test.loc[test['not_wear']>20]\n        test.sort_values('step_diff', inplace=True)\n        \n\n        test['day'] = ((test['step']//17280)+1).astype(np.int32)\n        \n        \n        mask_wk= (test['event']=='wakeup') #& (test['wakeup_zone']==1)\n        test_wakeup = test[mask_wk].groupby('day', as_index=False).agg('last')\n        \n        mask_os= (test['event']=='onset') #& (test['onset_zone']==1)\n        test_onset = test[mask_os].groupby('day', as_index=False).agg('last')\n\n        \n        df_events = pd.concat([test_wakeup, test_onset], ignore_index=True).sort_values('step')\n        \n        \n        if df_events.shape[0]>0:\n    \n            if df_events.iloc[0]['event'] == 'wakeup':\n                df_events = df_events.iloc[1:]\n\n    return df_events\n","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"%%time\n\ncols_sub = ['series_id','step','event','score']\n\nseries_id  = pd.read_parquet('/kaggle/input/child-mind-institute-detect-sleep-states/test_series.parquet', columns=['series_id'])\nseries_id = series_id.series_id.unique()\n\ntests = Parallel(n_jobs=-1)(delayed(get_events)(i, model) for i in series_id)","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Submission","metadata":{}},{"cell_type":"code","source":"submission = pd.concat(tests, ignore_index=True)\nsubmission = submission[cols_sub].reset_index(names='row_id')\nsubmission.to_csv('submission.csv', index=False)\nsubmission","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"# %%time\n\n# if DEV:\n    \n#     print('validation')\n\n#     series_id  = pd.read_parquet('/kaggle/input/child-mind-institute-detect-sleep-states/train_series.parquet', columns=['series_id'])\n#     series_id = series_id.series_id.unique()\n\n#     path = '/kaggle/input/child-mind-institute-detect-sleep-states/'\n\n#     train_events = pd.read_csv(path + 'train_events.csv')\n#     train_events['timestamp'] = pd.to_datetime(train_events['timestamp']).apply(lambda t: t.tz_localize(None))\n\n#     val_solution = train_events[train_events['series_id'].isin(series_id)]\n\n#     vals = Parallel(n_jobs=-1)(delayed(get_events)(i, model, file='train_series.parquet') for i in series_id)\n\n#     val=pd.concat(vals, ignore_index=True).reset_index(names='row_id')\n\n#     print(f\"LGBM score: {score(val_solution, val, tolerances, **column_names)}\")\n\n#     val['score'] = 1\n#     print(f\"LGBM score: {score(val_solution, val, tolerances, **column_names)}\")","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"","metadata":{},"execution_count":null,"outputs":[]}]}