{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.7.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"nvidiaTeslaT4","dataSources":[{"sourceId":31254,"databundleVersionId":3103714,"sourceType":"competition"},{"sourceId":3618498,"sourceType":"datasetVersion","datasetId":1931827},{"sourceId":94922230,"sourceType":"kernelVersion"}],"dockerImageVersionId":30153,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"%%time\nimport os\nimport sys\nimport copy\nfrom datetime import datetime\nimport gc\nimport pickle as pkl\nimport shelve\n\nimport pandas as pd\nimport numpy as np\nimport cudf\n    \nsys.path.append(\"../input/\")\nfrom handmhelpers import io as h_io, sub as h_sub, cv as h_cv, fe as h_fe\nfrom handmhelpers import modeling as h_modeling, candidates as h_can, pairs as h_pairs","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:20:46.494120Z","iopub.execute_input":"2025-05-23T09:20:46.494338Z","iopub.status.idle":"2025-05-23T09:20:55.733194Z","shell.execute_reply.started":"2025-05-23T09:20:46.494276Z","shell.execute_reply":"2025-05-23T09:20:55.732335Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"%%time\n\nc, t, a = h_io.load_data(files=['customers.csv', 'transactions_train.csv', 'articles.csv'])        \n\nindex_to_id_dict_path = h_fe.reduce_customer_id_memory(c, [t])\nt[\"week_number\"] = h_fe.day_week_numbers(t[\"t_dat\"])\nt[\"t_dat\"] = h_fe.day_numbers(t[\"t_dat\"])","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:21:00.522167Z","iopub.execute_input":"2025-05-23T09:21:00.522430Z","iopub.status.idle":"2025-05-23T09:22:04.150043Z","shell.execute_reply.started":"2025-05-23T09:21:00.522400Z","shell.execute_reply":"2025-05-23T09:22:04.149215Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"%%time\n\npairs_per_item = 5\n\nweek_number_pairs = {}\nfor week_number in [96, 97, 98, 99, 100, 101, 102, 103, 104]:\n    print(f\"Creating pairs for week number {week_number}\")\n    week_number_pairs[week_number] = h_pairs.create_pairs(\n        t, week_number, pairs_per_item, verbose=False\n    )","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:22:18.356862Z","iopub.execute_input":"2025-05-23T09:22:18.357545Z","iopub.status.idle":"2025-05-23T09:23:19.729319Z","shell.execute_reply.started":"2025-05-23T09:22:18.357515Z","shell.execute_reply":"2025-05-23T09:23:19.728411Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"def create_candidates_with_features_df(t, c, a, customer_batch=None, **kwargs):\n    features_df, label_df = h_cv.feature_label_split(\n        t, kwargs[\"label_week\"], kwargs[\"feature_periods\"]\n    )\n    \n    features_df[\"t_dat\"] = h_fe.how_many_ago(features_df[\"t_dat\"])\n    features_df[\"week_number\"] = h_fe.how_many_ago(features_df[\"week_number\"])\n    \n    article_pairs_df = week_number_pairs[kwargs[\"label_week\"]-1]\n    \n    if len(label_df) > 0:\n        customers = label_df[\"customer_id\"].unique()\n    elif customer_batch is not None:\n        customers = customer_batch\n    else:\n        customers = None\n    \n    features_db = shelve.open(\"features_db\") \n    \n    recent_customer_cand, features_db[\"customer_article\"] = (\n        h_can.create_recent_customer_candidates(\n            features_df,\n            kwargs[\"ca_num_weeks\"],\n            customers=customers,\n        )\n    )\n    \n    (cust_last_week_cand,\n     cust_last_week_pair_cand,\n     features_db[\"clw\"],\n     features_db[\"clw_pairs\"]) = h_can.create_last_customer_weeks_and_pairs(\n        features_df,\n        article_pairs_df,\n        kwargs[\"clw_num_weeks\"],\n        kwargs[\"clw_num_pair_weeks\"],\n        customers=customers,\n    )\n    \n    _, features_db[\"popular_articles\"] = h_can.create_popular_article_cand(\n        features_df,\n        c,\n        a,\n        kwargs[\"pa_num_weeks\"],\n        kwargs[\"hier_col\"],\n        num_candidates=kwargs[\"num_recent_candidates\"],\n        num_articles=kwargs[\"num_recent_articles\"],\n        customers=customers,\n    )\n    age_bucket_can, _, _ = h_can.create_age_bucket_candidates(\n        features_df,\n        c,\n        kwargs[\"num_age_buckets\"],\n        articles=kwargs[\"num_recent_articles\"],\n        customers=customers,\n    )\n    \n    cand = [recent_customer_cand, cust_last_week_cand, cust_last_week_pair_cand, age_bucket_can]\n    cand = cudf.concat(cand).drop_duplicates()\n    cand = cand.sort_values([\"customer_id\", \"article_id\"]).reset_index(drop=True)\n    \n    del recent_customer_cand, cust_last_week_cand, cust_last_week_pair_cand, age_bucket_can\n    \n    cand = h_can.filter_candidates(cand, t, **kwargs)\n    \n    h_fe.create_cust_hier_features(features_df, a, kwargs[\"hier_cols\"], features_db)\n    h_fe.create_price_features(features_df, features_db)\n    h_fe.create_cust_features(c, features_db)\n    h_fe.create_article_cust_features(features_df, c, features_db)\n    h_fe.create_lag_features(features_df, a, kwargs[\"lag_days\"], features_db)\n    h_fe.create_rebuy_features(features_df, features_db)\n    h_fe.create_cust_t_features(features_df, a, features_db)\n    h_fe.create_art_t_features(features_df, features_db)\n    \n    del features_df\n\n    if customers is not None:\n        cand = cand[cand[\"customer_id\"].isin(customers)]\n    \n    if kwargs[\"cv\"]:\n        ground_truth_candidates = label_df[[\"customer_id\", \"article_id\"]].drop_duplicates()\n        h_cv.report_candidates(cand, ground_truth_candidates)\n        del ground_truth_candidates        \n    \n    cand_with_f_df = h_can.add_features_to_candidates(\n        cand, features_db, c, a\n    )\n    \n    for article_col in kwargs[\"article_columns\"]:\n        art_col_map = a.set_index(\"article_id\")[article_col]\n        cand_with_f_df[article_col] = cand_with_f_df[\"article_id\"].map(art_col_map)\n    \n    if kwargs[\"selected_features\"] is not None:\n        cand_with_f_df = cand_with_f_df[\n            [\"customer_id\", \"article_id\"] + kwargs[\"selected_features\"]\n        ]\n        \n    features_db.close()\n    os.remove(\"features_db.bak\"), os.remove(\"features_db.dir\"), os.remove(\"features_db.dat\")\n    \n    assert len(cand) == len(cand_with_f_df), \"seem to have duplicates in the feature dfs\"\n    del cand\n    \n    return cand_with_f_df, label_df","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:23:41.736358Z","iopub.execute_input":"2025-05-23T09:23:41.736920Z","iopub.status.idle":"2025-05-23T09:23:41.751875Z","shell.execute_reply.started":"2025-05-23T09:23:41.736886Z","shell.execute_reply":"2025-05-23T09:23:41.751020Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"def calculate_model_score(ids_df, preds, truth_df):\n    predictions = h_modeling.create_predictions(ids_df, preds)\n    true_labels = h_cv.ground_truth(truth_df).set_index(\"customer_id\")[\"prediction\"]\n    score = round(h_cv.comp_average_precision(true_labels, predictions),5)\n    \n    return score","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:23:54.673605Z","iopub.execute_input":"2025-05-23T09:23:54.673864Z","iopub.status.idle":"2025-05-23T09:23:54.678317Z","shell.execute_reply.started":"2025-05-23T09:23:54.673834Z","shell.execute_reply":"2025-05-23T09:23:54.677596Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"cv_params = {\n    \"cv\": True,\n    \"feature_periods\": 105,\n    \"label_week\": 104,\n    \"index_to_id_dict_path\": index_to_id_dict_path,\n    \"pairs_file_version\": \"_v3_5_ex\",\n    \"num_recent_candidates\": 36,\n    \"num_recent_articles\": 12,\n    \"hier_col\": \"department_no\",\n    \"ca_num_weeks\": 3,\n    \"clw_num_weeks\": 12,\n    \"clw_num_pair_weeks\": 2,\n    \"pa_num_weeks\": 1,\n    \"num_age_buckets\": 4,\n    \"filter_recent_art_weeks\": 1,\n    \"filter_num_articles\": None,\n    \"lag_days\": [1, 3, 14, 30],\n    \"article_columns\": [\"index_code\"],\n    \"hier_cols\": [\n        \"department_no\", \"section_no\", \"index_group_no\", \"index_code\",\n        \"product_type_no\", \"product_group_name\"\n    ],\n    \"selected_features\": None,\n    \"lgbm_params\": {\"n_estimators\": 200, \"num_leaves\": 20},\n    \"log_evaluation\": 10,\n    \"early_stopping\": 20,\n    \"eval_at\": 12,\n    \"save_model\": True,\n    \"num_concats\": 5,\n}\nsub_params = {\n    \"cv\": False,\n    \"feature_periods\": 105,\n    \"label_week\": 105,\n    \"index_to_id_dict_path\": index_to_id_dict_path,\n    \"pairs_file_version\": \"_v3_5_ex\",\n    \"num_recent_candidates\": 60,\n    \"num_recent_articles\": 12,\n    \"hier_col\": \"department_no\",\n    \"ca_num_weeks\": 3,\n    \"clw_num_weeks\": 12,\n    \"clw_num_pair_weeks\": 2,\n    \"pa_num_weeks\": 1,\n    \"num_age_buckets\": 4,\n    \"filter_recent_art_weeks\": 1,\n    \"filter_num_articles\": None,\n    \"lag_days\": [1, 3, 14, 30],\n    \"article_columns\": [\"index_code\"],\n    \"hier_cols\": [\n        \"department_no\", \"section_no\", \"index_group_no\", \"index_code\",\n        \"product_type_no\", \"product_group_name\"\n    ],\n    \"selected_features\": None,\n    \"lgbm_params\": {\n        \"n_estimators\": 150,\n        \"num_leaves\": 20,    \n    },\n    \"log_evaluation\": 10,\n    \"eval_at\": 12,\n    \"prediction_models\": [\"model_104\", \"model_105\"],\n    \"save_model\": True,\n    \"num_concats\": 5,\n}","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:23:56.642338Z","iopub.execute_input":"2025-05-23T09:23:56.642913Z","iopub.status.idle":"2025-05-23T09:23:56.651440Z","shell.execute_reply.started":"2025-05-23T09:23:56.642876Z","shell.execute_reply":"2025-05-23T09:23:56.650668Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"cand_features_func = create_candidates_with_features_df\nscoring_func = calculate_model_score","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:24:12.502207Z","iopub.execute_input":"2025-05-23T09:24:12.502475Z","iopub.status.idle":"2025-05-23T09:24:12.506216Z","shell.execute_reply.started":"2025-05-23T09:24:12.502441Z","shell.execute_reply":"2025-05-23T09:24:12.505471Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"%%time\ncv_weeks = [104]\nresults = h_modeling.run_all_cvs(\n    t, c, a, cand_features_func, scoring_func, \n    cv_weeks=cv_weeks, **cv_params\n)","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:24:15.107035Z","iopub.execute_input":"2025-05-23T09:24:15.107269Z","iopub.status.idle":"2025-05-23T09:26:36.422645Z","shell.execute_reply.started":"2025-05-23T09:24:15.107246Z","shell.execute_reply":"2025-05-23T09:26:36.421925Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"%%time\ngc.collect()\nh_modeling.full_sub_train_run(t, c, a, cand_features_func, scoring_func, **sub_params)\npredictions = h_modeling.full_sub_predict_run(\n    t, c, a, cand_features_func, **sub_params\n)","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:29:14.293779Z","iopub.execute_input":"2025-05-23T09:29:14.294034Z","iopub.status.idle":"2025-05-23T09:35:56.158617Z","shell.execute_reply.started":"2025-05-23T09:29:14.294009Z","shell.execute_reply":"2025-05-23T09:35:56.157873Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"sub = h_sub.create_sub(c[\"customer_id\"], predictions, index_to_id_dict_path)\nsub.to_csv('dev_submission.csv', index=False)\n\ndisplay(sub.head())\nprint(sub.shape)","metadata":{"execution":{"iopub.status.busy":"2025-05-23T09:38:40.384547Z","iopub.execute_input":"2025-05-23T09:38:40.385287Z","iopub.status.idle":"2025-05-23T09:39:45.008407Z","shell.execute_reply.started":"2025-05-23T09:38:40.385254Z","shell.execute_reply":"2025-05-23T09:39:45.007681Z"},"trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"","metadata":{"trusted":true},"outputs":[],"execution_count":null}]}