baseline_by_project_weighted_by_round.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Baseline by project weighted by round (public score=0.1057)\n",
    "\n",
    "What has been done:\n",
    "- calculate funding amount from weight\n",
    "- calculate percentage of funding amount by project\n",
    "- normalize percentage by funder and quarter\n",
    "- use above funding amount to predict\n",
    "- null values are filled with 0.5"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "20958\n",
      "4261\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>id</th>\n",
       "      <th>project_a</th>\n",
       "      <th>project_b</th>\n",
       "      <th>weight_a</th>\n",
       "      <th>weight_b</th>\n",
       "      <th>total_amount_usd</th>\n",
       "      <th>funder</th>\n",
       "      <th>quarter</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>715</td>\n",
       "      <td>https://github.com/mochajs/mocha</td>\n",
       "      <td>https://github.com/chzyer/readline</td>\n",
       "      <td>0.961581</td>\n",
       "      <td>0.038419</td>\n",
       "      <td>2681</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2016-04</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>716</td>\n",
       "      <td>https://github.com/mochajs/mocha</td>\n",
       "      <td>https://github.com/gulpjs/gulp</td>\n",
       "      <td>0.992302</td>\n",
       "      <td>0.007698</td>\n",
       "      <td>2598</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2016-04</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>717</td>\n",
       "      <td>https://github.com/chzyer/readline</td>\n",
       "      <td>https://github.com/gulpjs/gulp</td>\n",
       "      <td>0.837398</td>\n",
       "      <td>0.162602</td>\n",
       "      <td>123</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2016-04</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>718</td>\n",
       "      <td>https://github.com/chzyer/readline</td>\n",
       "      <td>https://github.com/gulpjs/gulp</td>\n",
       "      <td>0.231595</td>\n",
       "      <td>0.768405</td>\n",
       "      <td>652</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2016-07</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>719</td>\n",
       "      <td>https://github.com/chzyer/readline</td>\n",
       "      <td>https://github.com/mochajs/mocha</td>\n",
       "      <td>0.078851</td>\n",
       "      <td>0.921149</td>\n",
       "      <td>1915</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2016-07</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "    id                           project_a  \\\n",
       "0  715    https://github.com/mochajs/mocha   \n",
       "1  716    https://github.com/mochajs/mocha   \n",
       "2  717  https://github.com/chzyer/readline   \n",
       "3  718  https://github.com/chzyer/readline   \n",
       "4  719  https://github.com/chzyer/readline   \n",
       "\n",
       "                            project_b  weight_a  weight_b  total_amount_usd  \\\n",
       "0  https://github.com/chzyer/readline  0.961581  0.038419              2681   \n",
       "1      https://github.com/gulpjs/gulp  0.992302  0.007698              2598   \n",
       "2      https://github.com/gulpjs/gulp  0.837398  0.162602               123   \n",
       "3      https://github.com/gulpjs/gulp  0.231595  0.768405               652   \n",
       "4    https://github.com/mochajs/mocha  0.078851  0.921149              1915   \n",
       "\n",
       "           funder  quarter  \n",
       "0  opencollective  2016-04  \n",
       "1  opencollective  2016-04  \n",
       "2  opencollective  2016-04  \n",
       "3  opencollective  2016-07  \n",
       "4  opencollective  2016-07  "
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "df = pd.read_csv('dataset/dataset.csv')\n",
    "test_df = pd.read_csv('dataset/test.csv')\n",
    "\n",
    "# dataset size\n",
    "print(len(df))\n",
    "print(len(test_df))\n",
    "\n",
    "df.head()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "104\n",
      "82\n",
      "117\n"
     ]
    }
   ],
   "source": [
    "# check unique projects (project_a and project_b intersect)\n",
    "\n",
    "df_nunique = set(df['project_a'].unique().tolist() + df['project_b'].unique().tolist())\n",
    "test_df_nunique = set(test_df['project_a'].unique().tolist() + test_df['project_b'].unique().tolist())\n",
    "\n",
    "total_unique = df_nunique.union(test_df_nunique)\n",
    "\n",
    "print(len(df_nunique))\n",
    "print(len(test_df_nunique))\n",
    "print(len(total_unique))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>id</th>\n",
       "      <th>project_a</th>\n",
       "      <th>project_b</th>\n",
       "      <th>total_amount_usd</th>\n",
       "      <th>funder</th>\n",
       "      <th>quarter</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>20884</td>\n",
       "      <td>https://github.com/rollup/rollup</td>\n",
       "      <td>https://github.com/webpack/webpack</td>\n",
       "      <td>29097</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2024-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>20885</td>\n",
       "      <td>https://github.com/rollup/rollup</td>\n",
       "      <td>https://github.com/babel/babel</td>\n",
       "      <td>26093</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2024-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>20886</td>\n",
       "      <td>https://github.com/rollup/rollup</td>\n",
       "      <td>https://github.com/sindresorhus/type-fest</td>\n",
       "      <td>4460</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2024-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>20887</td>\n",
       "      <td>https://github.com/rollup/rollup</td>\n",
       "      <td>https://github.com/inikulin/parse5</td>\n",
       "      <td>2454</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2024-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>20888</td>\n",
       "      <td>https://github.com/rollup/rollup</td>\n",
       "      <td>https://github.com/zloirock/core-js</td>\n",
       "      <td>13229</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>2024-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4247</th>\n",
       "      <td>25206</td>\n",
       "      <td>https://github.com/ethereum/go-ethereum</td>\n",
       "      <td>https://github.com/chainsafe/lodestar</td>\n",
       "      <td>572742</td>\n",
       "      <td>optimism</td>\n",
       "      <td>2024-10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4249</th>\n",
       "      <td>25208</td>\n",
       "      <td>https://github.com/ethereum/go-ethereum</td>\n",
       "      <td>https://github.com/bluealloy/revm</td>\n",
       "      <td>622843</td>\n",
       "      <td>optimism</td>\n",
       "      <td>2024-10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4251</th>\n",
       "      <td>25210</td>\n",
       "      <td>https://github.com/sigp/lighthouse</td>\n",
       "      <td>https://github.com/chainsafe/lodestar</td>\n",
       "      <td>411176</td>\n",
       "      <td>optimism</td>\n",
       "      <td>2024-10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4253</th>\n",
       "      <td>25212</td>\n",
       "      <td>https://github.com/sigp/lighthouse</td>\n",
       "      <td>https://github.com/bluealloy/revm</td>\n",
       "      <td>461277</td>\n",
       "      <td>optimism</td>\n",
       "      <td>2024-10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4256</th>\n",
       "      <td>25215</td>\n",
       "      <td>https://github.com/chainsafe/lodestar</td>\n",
       "      <td>https://github.com/bluealloy/revm</td>\n",
       "      <td>398133</td>\n",
       "      <td>optimism</td>\n",
       "      <td>2024-10</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>3865 rows × 6 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "         id                                project_a  \\\n",
       "0     20884         https://github.com/rollup/rollup   \n",
       "1     20885         https://github.com/rollup/rollup   \n",
       "2     20886         https://github.com/rollup/rollup   \n",
       "3     20887         https://github.com/rollup/rollup   \n",
       "4     20888         https://github.com/rollup/rollup   \n",
       "...     ...                                      ...   \n",
       "4247  25206  https://github.com/ethereum/go-ethereum   \n",
       "4249  25208  https://github.com/ethereum/go-ethereum   \n",
       "4251  25210       https://github.com/sigp/lighthouse   \n",
       "4253  25212       https://github.com/sigp/lighthouse   \n",
       "4256  25215    https://github.com/chainsafe/lodestar   \n",
       "\n",
       "                                      project_b  total_amount_usd  \\\n",
       "0            https://github.com/webpack/webpack             29097   \n",
       "1                https://github.com/babel/babel             26093   \n",
       "2     https://github.com/sindresorhus/type-fest              4460   \n",
       "3            https://github.com/inikulin/parse5              2454   \n",
       "4           https://github.com/zloirock/core-js             13229   \n",
       "...                                         ...               ...   \n",
       "4247      https://github.com/chainsafe/lodestar            572742   \n",
       "4249          https://github.com/bluealloy/revm            622843   \n",
       "4251      https://github.com/chainsafe/lodestar            411176   \n",
       "4253          https://github.com/bluealloy/revm            461277   \n",
       "4256          https://github.com/bluealloy/revm            398133   \n",
       "\n",
       "              funder  quarter  \n",
       "0     opencollective  2024-01  \n",
       "1     opencollective  2024-01  \n",
       "2     opencollective  2024-01  \n",
       "3     opencollective  2024-01  \n",
       "4     opencollective  2024-01  \n",
       "...              ...      ...  \n",
       "4247        optimism  2024-10  \n",
       "4249        optimism  2024-10  \n",
       "4251        optimism  2024-10  \n",
       "4253        optimism  2024-10  \n",
       "4256        optimism  2024-10  \n",
       "\n",
       "[3865 rows x 6 columns]"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# check if there are any row in test_df that both project_a and project_b are in df\n",
    "test_df[test_df['project_a'].isin(df_nunique) & test_df['project_b'].isin(df_nunique)]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "# calculate funding amount\n",
    "\n",
    "df['amount_a'] = df['weight_a'] * df['total_amount_usd']\n",
    "df['amount_b'] = df['weight_b'] * df['total_amount_usd']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>quarter</th>\n",
       "      <th>funder</th>\n",
       "      <th>project</th>\n",
       "      <th>amount</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2016-04</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/chzyer/readline</td>\n",
       "      <td>103.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2016-04</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/gulpjs/gulp</td>\n",
       "      <td>20.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2016-04</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/mochajs/mocha</td>\n",
       "      <td>2578.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2016-07</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/chzyer/readline</td>\n",
       "      <td>151.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2016-07</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/gulpjs/gulp</td>\n",
       "      <td>501.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1149</th>\n",
       "      <td>2023-10</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/webdriverio/webdriverio</td>\n",
       "      <td>135.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1150</th>\n",
       "      <td>2023-10</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/webpack/webpack</td>\n",
       "      <td>25140.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1151</th>\n",
       "      <td>2023-10</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/webreflection/flatted</td>\n",
       "      <td>711.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1152</th>\n",
       "      <td>2023-10</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/yarnpkg/yarn</td>\n",
       "      <td>1047.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1153</th>\n",
       "      <td>2023-10</td>\n",
       "      <td>opencollective</td>\n",
       "      <td>https://github.com/zloirock/core-js</td>\n",
       "      <td>12126.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>1154 rows × 4 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      quarter          funder                                     project  \\\n",
       "0     2016-04  opencollective          https://github.com/chzyer/readline   \n",
       "1     2016-04  opencollective              https://github.com/gulpjs/gulp   \n",
       "2     2016-04  opencollective            https://github.com/mochajs/mocha   \n",
       "3     2016-07  opencollective          https://github.com/chzyer/readline   \n",
       "4     2016-07  opencollective              https://github.com/gulpjs/gulp   \n",
       "...       ...             ...                                         ...   \n",
       "1149  2023-10  opencollective  https://github.com/webdriverio/webdriverio   \n",
       "1150  2023-10  opencollective          https://github.com/webpack/webpack   \n",
       "1151  2023-10  opencollective    https://github.com/webreflection/flatted   \n",
       "1152  2023-10  opencollective             https://github.com/yarnpkg/yarn   \n",
       "1153  2023-10  opencollective         https://github.com/zloirock/core-js   \n",
       "\n",
       "       amount  \n",
       "0       103.0  \n",
       "1        20.0  \n",
       "2      2578.0  \n",
       "3       151.0  \n",
       "4       501.0  \n",
       "...       ...  \n",
       "1149    135.0  \n",
       "1150  25140.0  \n",
       "1151    711.0  \n",
       "1152   1047.0  \n",
       "1153  12126.0  \n",
       "\n",
       "[1154 rows x 4 columns]"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# extract unique project_a and amount_a, project_b and amount_b per quarter and funder, and then unfold\n",
    "\n",
    "df_a = df.groupby(['quarter', 'funder', 'project_a'])['amount_a'].mean().reset_index()\n",
    "df_b = df.groupby(['quarter', 'funder', 'project_b'])['amount_b'].mean().reset_index()\n",
    "\n",
    "# rename project_a and project_b to project\n",
    "df_a.rename(columns={'project_a': 'project'}, inplace=True)\n",
    "df_b.rename(columns={'project_b': 'project'}, inplace=True)\n",
    "\n",
    "# rename amount_a and amount_b to amount\n",
    "df_a.rename(columns={'amount_a': 'amount'}, inplace=True)\n",
    "df_b.rename(columns={'amount_b': 'amount'}, inplace=True)\n",
    "\n",
    "# concat df_a and df_b\n",
    "df_ab = pd.concat([df_a, df_b]).groupby(['quarter', 'funder', 'project']).mean().reset_index()\n",
    "\n",
    "df_ab"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "# divide amount by total amount got per project\n",
    "df_ab['pct_amount'] = df_ab['amount'] / df_ab['project'].map(df_ab.groupby(['project']).amount.sum())\n",
    "\n",
    "# average pct_amount by funder and quarter\n",
    "df_ab_avg = df_ab.groupby(['funder', 'quarter']).pct_amount.mean()\n",
    "\n",
    "# normalize amount by funder and quarter\n",
    "df_ab['scaled_amount'] = df_ab['amount'] * df_ab.apply(lambda row: df_ab_avg.get((row['funder'], row['quarter']), np.nan), axis=1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<class 'pandas.core.frame.DataFrame'>\n",
      "RangeIndex: 4261 entries, 0 to 4260\n",
      "Data columns (total 8 columns):\n",
      " #   Column            Non-Null Count  Dtype  \n",
      "---  ------            --------------  -----  \n",
      " 0   id                4261 non-null   int64  \n",
      " 1   project_a         4261 non-null   object \n",
      " 2   project_b         4261 non-null   object \n",
      " 3   total_amount_usd  4261 non-null   int64  \n",
      " 4   funder            4261 non-null   object \n",
      " 5   quarter           4261 non-null   object \n",
      " 6   score_a           4011 non-null   float64\n",
      " 7   score_b           4081 non-null   float64\n",
      "dtypes: float64(2), int64(2), object(4)\n",
      "memory usage: 266.4+ KB\n"
     ]
    }
   ],
   "source": [
    "# approach by doing one score per project\n",
    "fd = df_ab.groupby(['project']).scaled_amount.sum()\n",
    "\n",
    "test_fd = test_df.copy()\n",
    "\n",
    "# fill amount into test_fd by looking up project_a and project_b\n",
    "test_fd['score_a'] = test_fd['project_a'].map(fd)\n",
    "test_fd['score_b'] = test_fd['project_b'].map(fd)\n",
    "\n",
    "test_fd.info()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/var/folders/gt/sg3v8rd13l52jx91mfbgzbfc0000gn/T/ipykernel_98395/235156080.py:6: FutureWarning: A value is trying to be set on a copy of a DataFrame or Series through chained assignment using an inplace method.\n",
      "The behavior will change in pandas 3.0. This inplace method will never work because the intermediate object on which we are setting values always behaves as a copy.\n",
      "\n",
      "For example, when doing 'df[col].method(value, inplace=True)', try using 'df.method({col: value}, inplace=True)' or df[col] = df[col].method(value) instead, to perform the operation inplace on the original object.\n",
      "\n",
      "\n",
      "  test_fd['pred'].fillna(0.5, inplace=True)\n"
     ]
    }
   ],
   "source": [
    "# calculate pred (weight_a by score_a / (score_a + score_b))\n",
    "\n",
    "test_fd['pred'] = test_fd['score_a'] / (test_fd['score_a'] + test_fd['score_b'])\n",
    "\n",
    "# fill in null values with 0.5\n",
    "test_fd['pred'].fillna(0.5, inplace=True)\n",
    "\n",
    "# save id, pred to csv\n",
    "test_fd[['id', 'pred']].to_csv('submissions/baseline_by_project_weighted_by_round.csv', index=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "np.float64(0.07336304058389455)"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# check against train set\n",
    "\n",
    "df['pred_amount_a'] = df['project_a'].map(fd)\n",
    "df['pred_amount_b'] = df['project_b'].map(fd)\n",
    "\n",
    "df['pred'] = df['pred_amount_a'] / (df['pred_amount_a'] + df['pred_amount_b'])\n",
    "\n",
    "# calculate MSE between pred and weight_a\n",
    "\n",
    "df['mse'] = (df['pred'] - df['weight_a']) ** 2\n",
    "\n",
    "df['mse'].mean()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "xgboost",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.13.1"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}