Link_prediction_using_GraphSAGE.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "Copy of pilot_KG_Embedd_entities_allfeat.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "toc_visible": true,
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/Rishujamaiyar/desktop-tutorial/blob/master/Copy_of_pilot_KG_Embedd_entities_allfeat.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "3gChVhyM6f_D",
        "outputId": "3ea8b362-c80b-439b-83fe-24288cd22396",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "source": [
        "import sys\n",
        "if 'google.colab' in sys.modules:\n",
        "  %pip install -q stellargraph[demos]==1.0.0rc1\n",
        "\n",
        "  # verify that we're using the correct version of StellarGraph for this notebook\n",
        "import stellargraph as sg\n",
        "\n",
        "try:\n",
        "    sg.utils.validate_notebook_version(\"1.0.0rc1\")\n",
        "except AttributeError:\n",
        "    raise ValueError(\n",
        "        f\"This notebook requires StellarGraph version 1.0.0rc1, but a different version {sg.__version__} is installed.  Please see <https://github.com/stellargraph/stellargraph/issues/1172>.\"\n",
        "    ) from None"
      ],
      "execution_count": 1,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "\u001b[K     |████████████████████████████████| 378kB 2.6MB/s \n",
            "\u001b[K     |████████████████████████████████| 235kB 13.1MB/s \n",
            "\u001b[K     |████████████████████████████████| 51kB 6.6MB/s \n",
            "\u001b[?25h  Building wheel for mplleaflet (setup.py) ... \u001b[?25l\u001b[?25hdone\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "uZfkOHEb6zNj"
      },
      "source": [
        "import pandas as pd\n",
        "import numpy as np\n",
        "import random\n",
        "\n",
        "from stellargraph.mapper import GraphSAGELinkGenerator\n",
        "from stellargraph.mapper import GraphSAGENodeGenerator\n",
        "from stellargraph.layer import GraphSAGE, link_classification\n",
        "from stellargraph.data import UniformRandomWalk\n",
        "from stellargraph.data import UnsupervisedSampler"
      ],
      "execution_count": 74,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "He0vsDhmQTZm",
        "outputId": "454b797d-9a2f-4369-adc5-0f1a3244a5c1",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 424
        }
      },
      "source": [
        "nodelink=pd.read_csv('/content/ckt_links.csv',names=['source','target'])\n",
        "nodelink=nodelink.drop(nodelink.index[0])\n",
        "nodelink"
      ],
      "execution_count": 75,
      "outputs": [
        {
          "output_type": "execute_result",
          "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>source</th>\n",
              "      <th>target</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>Kedar Jadhav</td>\n",
              "      <td>Bat_IND</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>Mayank Agarwal</td>\n",
              "      <td>Bat_IND</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>Rohit Sharma</td>\n",
              "      <td>Bat_IND</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>Shikhar Dhawan</td>\n",
              "      <td>Bat_IND</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>5</th>\n",
              "      <td>Virat Kohli</td>\n",
              "      <td>Bat_IND</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>196</th>\n",
              "      <td>SL</td>\n",
              "      <td>WC</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>197</th>\n",
              "      <td>SA</td>\n",
              "      <td>WC</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>198</th>\n",
              "      <td>BAN</td>\n",
              "      <td>WC</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>199</th>\n",
              "      <td>WI</td>\n",
              "      <td>WC</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>200</th>\n",
              "      <td>AFG</td>\n",
              "      <td>WC</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>200 rows × 2 columns</p>\n",
              "</div>"
            ],
            "text/plain": [
              "             source   target\n",
              "1      Kedar Jadhav  Bat_IND\n",
              "2    Mayank Agarwal  Bat_IND\n",
              "3      Rohit Sharma  Bat_IND\n",
              "4    Shikhar Dhawan  Bat_IND\n",
              "5       Virat Kohli  Bat_IND\n",
              "..              ...      ...\n",
              "196              SL       WC\n",
              "197              SA       WC\n",
              "198             BAN       WC\n",
              "199              WI       WC\n",
              "200             AFG       WC\n",
              "\n",
              "[200 rows x 2 columns]"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 75
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "xIoU_7PlSrM9",
        "outputId": "2ff1c4ef-2123-4cae-fcbd-348c67d39d10",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 562
        }
      },
      "source": [
        "nodefeat=pd.read_csv('/content/ckt_feat.csv')\n",
        "nodefeat=nodefeat.set_index('players')\n",
        "nodefeat_raw = nodefeat\n",
        "nodefeat=pd.get_dummies(nodefeat,columns=['feat'])\n",
        "nodefeat"
      ],
      "execution_count": 76,
      "outputs": [
        {
          "output_type": "execute_result",
          "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>feat_1</th>\n",
              "      <th>feat_2</th>\n",
              "      <th>feat_3</th>\n",
              "      <th>feat_4</th>\n",
              "      <th>feat_5</th>\n",
              "      <th>feat_6</th>\n",
              "      <th>feat_7</th>\n",
              "      <th>feat_8</th>\n",
              "      <th>feat_9</th>\n",
              "      <th>feat_10</th>\n",
              "      <th>feat_11</th>\n",
              "      <th>feat_12</th>\n",
              "      <th>feat_13</th>\n",
              "      <th>feat_14</th>\n",
              "      <th>feat_15</th>\n",
              "      <th>feat_16</th>\n",
              "      <th>feat_17</th>\n",
              "      <th>feat_18</th>\n",
              "      <th>feat_19</th>\n",
              "      <th>feat_20</th>\n",
              "      <th>feat_21</th>\n",
              "      <th>feat_22</th>\n",
              "      <th>feat_23</th>\n",
              "      <th>feat_24</th>\n",
              "      <th>feat_25</th>\n",
              "      <th>feat_26</th>\n",
              "      <th>feat_27</th>\n",
              "      <th>feat_28</th>\n",
              "      <th>feat_29</th>\n",
              "      <th>feat_30</th>\n",
              "      <th>feat_31</th>\n",
              "      <th>feat_32</th>\n",
              "      <th>feat_33</th>\n",
              "      <th>feat_34</th>\n",
              "      <th>feat_35</th>\n",
              "      <th>feat_36</th>\n",
              "      <th>feat_37</th>\n",
              "      <th>feat_38</th>\n",
              "      <th>feat_39</th>\n",
              "      <th>feat_40</th>\n",
              "      <th>...</th>\n",
              "      <th>feat_162</th>\n",
              "      <th>feat_163</th>\n",
              "      <th>feat_164</th>\n",
              "      <th>feat_165</th>\n",
              "      <th>feat_166</th>\n",
              "      <th>feat_167</th>\n",
              "      <th>feat_168</th>\n",
              "      <th>feat_169</th>\n",
              "      <th>feat_170</th>\n",
              "      <th>feat_171</th>\n",
              "      <th>feat_172</th>\n",
              "      <th>feat_173</th>\n",
              "      <th>feat_174</th>\n",
              "      <th>feat_175</th>\n",
              "      <th>feat_176</th>\n",
              "      <th>feat_177</th>\n",
              "      <th>feat_178</th>\n",
              "      <th>feat_179</th>\n",
              "      <th>feat_180</th>\n",
              "      <th>feat_181</th>\n",
              "      <th>feat_182</th>\n",
              "      <th>feat_183</th>\n",
              "      <th>feat_184</th>\n",
              "      <th>feat_185</th>\n",
              "      <th>feat_186</th>\n",
              "      <th>feat_187</th>\n",
              "      <th>feat_188</th>\n",
              "      <th>feat_189</th>\n",
              "      <th>feat_190</th>\n",
              "      <th>feat_191</th>\n",
              "      <th>feat_192</th>\n",
              "      <th>feat_193</th>\n",
              "      <th>feat_194</th>\n",
              "      <th>feat_195</th>\n",
              "      <th>feat_196</th>\n",
              "      <th>feat_197</th>\n",
              "      <th>feat_198</th>\n",
              "      <th>feat_199</th>\n",
              "      <th>feat_200</th>\n",
              "      <th>feat_201</th>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>players</th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>Kedar Jadhav</th>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Mayank Agarwal</th>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Rohit Sharma</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Shikhar Dhawan</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Virat Kohli</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</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",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>SA</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>BAN</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>WI</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>AFG</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>WC</th>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>...</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>201 rows × 201 columns</p>\n",
              "</div>"
            ],
            "text/plain": [
              "                feat_1  feat_2  feat_3  ...  feat_199  feat_200  feat_201\n",
              "players                                 ...                              \n",
              "Kedar Jadhav         1       0       0  ...         0         0         0\n",
              "Mayank Agarwal       0       1       0  ...         0         0         0\n",
              "Rohit Sharma         0       0       1  ...         0         0         0\n",
              "Shikhar Dhawan       0       0       0  ...         0         0         0\n",
              "Virat Kohli          0       0       0  ...         0         0         0\n",
              "...                ...     ...     ...  ...       ...       ...       ...\n",
              "SA                   0       0       0  ...         0         0         0\n",
              "BAN                  0       0       0  ...         0         0         0\n",
              "WI                   0       0       0  ...         1         0         0\n",
              "AFG                  0       0       0  ...         0         1         0\n",
              "WC                   0       0       0  ...         0         0         1\n",
              "\n",
              "[201 rows x 201 columns]"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 76
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "xpXSSr64feZ0",
        "outputId": "8592fc23-7c7b-440d-a520-f13a1bd37107",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "source": [
        "from stellargraph import StellarGraph\n",
        "datagraph = StellarGraph( {\"source\": nodefeat}, {\"line\": nodelink})\n",
        "print(datagraph.info())"
      ],
      "execution_count": 78,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "StellarGraph: Undirected multigraph\n",
            " Nodes: 201, Edges: 200\n",
            "\n",
            " Node types:\n",
            "  source: [201]\n",
            "    Features: float32 vector, length 201\n",
            "    Edge types: source-line->source\n",
            "\n",
            " Edge types:\n",
            "    source-line->source: [200]\n",
            "        Weights: all 1 (default)\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "PfSAk3OtVIxz"
      },
      "source": [
        "#Unsupervised GraphSAGE implementation starts here\n",
        "\n",
        "nodes = list(datagraph.nodes())\n",
        "number_of_walks = 5\n",
        "length = 4\n",
        "batch_size = 10\n",
        "epochs = 5\n",
        "num_samples = [5, 4]"
      ],
      "execution_count": 79,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "LRwcYEJuVdCB"
      },
      "source": [
        "unsupervised_samples = UnsupervisedSampler(datagraph, nodes=nodes, length=length, number_of_walks=number_of_walks)"
      ],
      "execution_count": 80,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "5alcqtHuVZOx"
      },
      "source": [
        "generator = GraphSAGELinkGenerator(datagraph, batch_size, num_samples)\n",
        "train_gen = generator.flow(unsupervised_samples)"
      ],
      "execution_count": 81,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "sirczPg2WQ1t"
      },
      "source": [
        "layer_sizes = [40, 40]            #determine the size of layers and hence output vectors\n",
        "graphsage = GraphSAGE(layer_sizes=layer_sizes, generator=generator, bias=True, dropout=0.0, normalize=\"l2\")"
      ],
      "execution_count": 82,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "_bMR8x5xWTBj",
        "outputId": "e5fd11f6-3fca-49f9-b1f7-dd51116b12bf",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "source": [
        "# Build the model and expose input and output sockets of graphsage, for node pair inputs:\n",
        "x_inp, x_out = graphsage.in_out_tensors()\n",
        "prediction = link_classification(output_dim=1, output_act=\"sigmoid\", edge_embedding_method=\"ip\")(x_out)"
      ],
      "execution_count": 83,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "link_classification: using 'ip' method to combine node embeddings into edge embeddings\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "kDZKk3ykWbuK"
      },
      "source": [
        "model = keras.Model(inputs=x_inp, outputs=prediction)\n",
        "\n",
        "model.compile(\n",
        "    optimizer=keras.optimizers.Adam(lr=1e-3),\n",
        "    loss=keras.losses.binary_crossentropy,\n",
        "    metrics=[keras.metrics.binary_accuracy],\n",
        ")"
      ],
      "execution_count": 84,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "ifr1eMr3Wgvs",
        "outputId": "42116882-9ee3-4d2f-b087-d36608adbd06",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "source": [
        "history = model.fit(\n",
        "    train_gen,\n",
        "    epochs=epochs,\n",
        "    verbose=1,\n",
        "    use_multiprocessing=False,\n",
        "    workers=4,\n",
        "    shuffle=True,\n",
        ")"
      ],
      "execution_count": 85,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "Epoch 1/5\n",
            "603/603 [==============================] - 6s 10ms/step - loss: 0.5834 - binary_accuracy: 0.7166\n",
            "Epoch 2/5\n",
            "603/603 [==============================] - 6s 10ms/step - loss: 0.5397 - binary_accuracy: 0.7474\n",
            "Epoch 3/5\n",
            "603/603 [==============================] - 6s 10ms/step - loss: 0.5363 - binary_accuracy: 0.7570\n",
            "Epoch 4/5\n",
            "603/603 [==============================] - 6s 10ms/step - loss: 0.5251 - binary_accuracy: 0.7794\n",
            "Epoch 5/5\n",
            "603/603 [==============================] - 6s 10ms/step - loss: 0.5248 - binary_accuracy: 0.7867\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "SFhXAVjiWrHE"
      },
      "source": [
        "x_inp_src = x_inp[0::2]\n",
        "x_out_src = x_out[0]\n",
        "embedding_model = keras.Model(inputs=x_inp_src, outputs=x_out_src)"
      ],
      "execution_count": 86,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "VfBLeNn0Wsps"
      },
      "source": [
        "node_ids = nodefeat.index\n",
        "node_gen = GraphSAGENodeGenerator(datagraph, batch_size, num_samples).flow(node_ids)"
      ],
      "execution_count": 87,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "G3LCR2jmXHQ0",
        "outputId": "c8bff7c8-3f9e-4f0b-e610-168a9f409c2a",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "source": [
        "node_embeddings = embedding_model.predict(node_gen, workers=4, verbose=1)"
      ],
      "execution_count": 88,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "21/21 [==============================] - 0s 7ms/step\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "c-uyQMzRan7X",
        "outputId": "facadab1-eab3-4283-dbe7-ca683a9635c0",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 562
        }
      },
      "source": [
        "node_embedd=pd.DataFrame(node_embeddings)\n",
        "node_embedd['nodes']=nodes\n",
        "node_embedd=node_embedd.set_index('nodes')\n",
        "node_embedd"
      ],
      "execution_count": 89,
      "outputs": [
        {
          "output_type": "execute_result",
          "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>0</th>\n",
              "      <th>1</th>\n",
              "      <th>2</th>\n",
              "      <th>3</th>\n",
              "      <th>4</th>\n",
              "      <th>5</th>\n",
              "      <th>6</th>\n",
              "      <th>7</th>\n",
              "      <th>8</th>\n",
              "      <th>9</th>\n",
              "      <th>10</th>\n",
              "      <th>11</th>\n",
              "      <th>12</th>\n",
              "      <th>13</th>\n",
              "      <th>14</th>\n",
              "      <th>15</th>\n",
              "      <th>16</th>\n",
              "      <th>17</th>\n",
              "      <th>18</th>\n",
              "      <th>19</th>\n",
              "      <th>20</th>\n",
              "      <th>21</th>\n",
              "      <th>22</th>\n",
              "      <th>23</th>\n",
              "      <th>24</th>\n",
              "      <th>25</th>\n",
              "      <th>26</th>\n",
              "      <th>27</th>\n",
              "      <th>28</th>\n",
              "      <th>29</th>\n",
              "      <th>30</th>\n",
              "      <th>31</th>\n",
              "      <th>32</th>\n",
              "      <th>33</th>\n",
              "      <th>34</th>\n",
              "      <th>35</th>\n",
              "      <th>36</th>\n",
              "      <th>37</th>\n",
              "      <th>38</th>\n",
              "      <th>39</th>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>nodes</th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "      <th></th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>Kedar Jadhav</th>\n",
              "      <td>0.292280</td>\n",
              "      <td>0.056647</td>\n",
              "      <td>-0.262089</td>\n",
              "      <td>0.003980</td>\n",
              "      <td>-0.076056</td>\n",
              "      <td>0.052593</td>\n",
              "      <td>-0.055122</td>\n",
              "      <td>-0.101379</td>\n",
              "      <td>-0.206529</td>\n",
              "      <td>-0.064279</td>\n",
              "      <td>-0.017356</td>\n",
              "      <td>0.097088</td>\n",
              "      <td>0.159895</td>\n",
              "      <td>0.037582</td>\n",
              "      <td>-0.065818</td>\n",
              "      <td>0.160330</td>\n",
              "      <td>0.010954</td>\n",
              "      <td>-0.173763</td>\n",
              "      <td>-0.079958</td>\n",
              "      <td>0.097588</td>\n",
              "      <td>-0.160301</td>\n",
              "      <td>-0.018526</td>\n",
              "      <td>-0.187540</td>\n",
              "      <td>-0.021803</td>\n",
              "      <td>-0.101526</td>\n",
              "      <td>-0.184155</td>\n",
              "      <td>0.158137</td>\n",
              "      <td>0.390427</td>\n",
              "      <td>0.158838</td>\n",
              "      <td>-0.009203</td>\n",
              "      <td>-0.201514</td>\n",
              "      <td>0.154706</td>\n",
              "      <td>-0.022875</td>\n",
              "      <td>0.114511</td>\n",
              "      <td>0.109329</td>\n",
              "      <td>0.313659</td>\n",
              "      <td>0.276787</td>\n",
              "      <td>-0.278417</td>\n",
              "      <td>-0.054648</td>\n",
              "      <td>0.082107</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Mayank Agarwal</th>\n",
              "      <td>0.272155</td>\n",
              "      <td>0.099263</td>\n",
              "      <td>-0.243889</td>\n",
              "      <td>0.013179</td>\n",
              "      <td>-0.127486</td>\n",
              "      <td>0.029820</td>\n",
              "      <td>-0.013500</td>\n",
              "      <td>-0.106994</td>\n",
              "      <td>-0.182825</td>\n",
              "      <td>-0.045790</td>\n",
              "      <td>0.028234</td>\n",
              "      <td>0.082182</td>\n",
              "      <td>0.203127</td>\n",
              "      <td>0.062685</td>\n",
              "      <td>-0.092170</td>\n",
              "      <td>0.102560</td>\n",
              "      <td>0.000609</td>\n",
              "      <td>-0.191952</td>\n",
              "      <td>0.003946</td>\n",
              "      <td>0.070629</td>\n",
              "      <td>-0.153807</td>\n",
              "      <td>-0.030444</td>\n",
              "      <td>-0.189957</td>\n",
              "      <td>-0.003821</td>\n",
              "      <td>-0.121522</td>\n",
              "      <td>-0.182517</td>\n",
              "      <td>0.142391</td>\n",
              "      <td>0.388737</td>\n",
              "      <td>0.159589</td>\n",
              "      <td>0.004919</td>\n",
              "      <td>-0.243273</td>\n",
              "      <td>0.136858</td>\n",
              "      <td>-0.022217</td>\n",
              "      <td>0.119198</td>\n",
              "      <td>0.102117</td>\n",
              "      <td>0.303017</td>\n",
              "      <td>0.283181</td>\n",
              "      <td>-0.286260</td>\n",
              "      <td>-0.052138</td>\n",
              "      <td>0.114731</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Rohit Sharma</th>\n",
              "      <td>0.316569</td>\n",
              "      <td>0.044673</td>\n",
              "      <td>-0.267582</td>\n",
              "      <td>0.032701</td>\n",
              "      <td>-0.063212</td>\n",
              "      <td>0.085152</td>\n",
              "      <td>-0.047418</td>\n",
              "      <td>-0.094766</td>\n",
              "      <td>-0.221316</td>\n",
              "      <td>-0.048173</td>\n",
              "      <td>-0.040861</td>\n",
              "      <td>0.097376</td>\n",
              "      <td>0.160200</td>\n",
              "      <td>0.051717</td>\n",
              "      <td>-0.070390</td>\n",
              "      <td>0.147122</td>\n",
              "      <td>0.043633</td>\n",
              "      <td>-0.212105</td>\n",
              "      <td>-0.075532</td>\n",
              "      <td>0.088612</td>\n",
              "      <td>-0.152574</td>\n",
              "      <td>-0.034476</td>\n",
              "      <td>-0.175623</td>\n",
              "      <td>-0.012666</td>\n",
              "      <td>-0.110974</td>\n",
              "      <td>-0.177349</td>\n",
              "      <td>0.135329</td>\n",
              "      <td>0.364753</td>\n",
              "      <td>0.168255</td>\n",
              "      <td>-0.004148</td>\n",
              "      <td>-0.230888</td>\n",
              "      <td>0.133306</td>\n",
              "      <td>-0.009844</td>\n",
              "      <td>0.100340</td>\n",
              "      <td>0.099278</td>\n",
              "      <td>0.309573</td>\n",
              "      <td>0.268754</td>\n",
              "      <td>-0.271688</td>\n",
              "      <td>-0.040246</td>\n",
              "      <td>0.106200</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Shikhar Dhawan</th>\n",
              "      <td>0.269974</td>\n",
              "      <td>0.045036</td>\n",
              "      <td>-0.268864</td>\n",
              "      <td>0.014538</td>\n",
              "      <td>-0.083343</td>\n",
              "      <td>0.073109</td>\n",
              "      <td>-0.056096</td>\n",
              "      <td>-0.098112</td>\n",
              "      <td>-0.200035</td>\n",
              "      <td>-0.046676</td>\n",
              "      <td>-0.021854</td>\n",
              "      <td>0.082994</td>\n",
              "      <td>0.136230</td>\n",
              "      <td>0.056643</td>\n",
              "      <td>-0.083025</td>\n",
              "      <td>0.154360</td>\n",
              "      <td>0.067142</td>\n",
              "      <td>-0.205017</td>\n",
              "      <td>-0.103812</td>\n",
              "      <td>0.110550</td>\n",
              "      <td>-0.150447</td>\n",
              "      <td>-0.028549</td>\n",
              "      <td>-0.191436</td>\n",
              "      <td>-0.003748</td>\n",
              "      <td>-0.115073</td>\n",
              "      <td>-0.180498</td>\n",
              "      <td>0.143140</td>\n",
              "      <td>0.365969</td>\n",
              "      <td>0.170043</td>\n",
              "      <td>-0.001188</td>\n",
              "      <td>-0.238305</td>\n",
              "      <td>0.149632</td>\n",
              "      <td>-0.013037</td>\n",
              "      <td>0.097904</td>\n",
              "      <td>0.099155</td>\n",
              "      <td>0.299887</td>\n",
              "      <td>0.284178</td>\n",
              "      <td>-0.288133</td>\n",
              "      <td>-0.051525</td>\n",
              "      <td>0.107410</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>Virat Kohli</th>\n",
              "      <td>0.305277</td>\n",
              "      <td>0.046292</td>\n",
              "      <td>-0.283788</td>\n",
              "      <td>0.004528</td>\n",
              "      <td>-0.087205</td>\n",
              "      <td>0.042213</td>\n",
              "      <td>-0.022424</td>\n",
              "      <td>-0.077621</td>\n",
              "      <td>-0.202520</td>\n",
              "      <td>-0.050006</td>\n",
              "      <td>0.000516</td>\n",
              "      <td>0.069855</td>\n",
              "      <td>0.207337</td>\n",
              "      <td>0.048577</td>\n",
              "      <td>-0.078940</td>\n",
              "      <td>0.099965</td>\n",
              "      <td>0.027705</td>\n",
              "      <td>-0.217809</td>\n",
              "      <td>-0.037599</td>\n",
              "      <td>0.123332</td>\n",
              "      <td>-0.173918</td>\n",
              "      <td>-0.040165</td>\n",
              "      <td>-0.191621</td>\n",
              "      <td>-0.034143</td>\n",
              "      <td>-0.071609</td>\n",
              "      <td>-0.151755</td>\n",
              "      <td>0.139150</td>\n",
              "      <td>0.361397</td>\n",
              "      <td>0.177463</td>\n",
              "      <td>0.004297</td>\n",
              "      <td>-0.250108</td>\n",
              "      <td>0.143981</td>\n",
              "      <td>0.017548</td>\n",
              "      <td>0.083265</td>\n",
              "      <td>0.089759</td>\n",
              "      <td>0.306650</td>\n",
              "      <td>0.270497</td>\n",
              "      <td>-0.265505</td>\n",
              "      <td>-0.044504</td>\n",
              "      <td>0.125493</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",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>SA</th>\n",
              "      <td>-0.084763</td>\n",
              "      <td>-0.113558</td>\n",
              "      <td>-0.083624</td>\n",
              "      <td>0.009571</td>\n",
              "      <td>-0.030005</td>\n",
              "      <td>0.097219</td>\n",
              "      <td>0.089951</td>\n",
              "      <td>-0.080451</td>\n",
              "      <td>0.135462</td>\n",
              "      <td>0.132102</td>\n",
              "      <td>-0.047145</td>\n",
              "      <td>-0.164905</td>\n",
              "      <td>-0.095662</td>\n",
              "      <td>0.152843</td>\n",
              "      <td>0.021036</td>\n",
              "      <td>-0.163975</td>\n",
              "      <td>0.170557</td>\n",
              "      <td>0.215077</td>\n",
              "      <td>-0.094512</td>\n",
              "      <td>0.208128</td>\n",
              "      <td>0.084350</td>\n",
              "      <td>0.073653</td>\n",
              "      <td>-0.007869</td>\n",
              "      <td>-0.099774</td>\n",
              "      <td>0.208735</td>\n",
              "      <td>-0.090742</td>\n",
              "      <td>-0.100339</td>\n",
              "      <td>-0.243725</td>\n",
              "      <td>-0.130537</td>\n",
              "      <td>-0.330090</td>\n",
              "      <td>0.275832</td>\n",
              "      <td>-0.244942</td>\n",
              "      <td>0.226957</td>\n",
              "      <td>0.163737</td>\n",
              "      <td>0.135433</td>\n",
              "      <td>0.035577</td>\n",
              "      <td>-0.180913</td>\n",
              "      <td>0.147233</td>\n",
              "      <td>-0.172231</td>\n",
              "      <td>-0.327650</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>BAN</th>\n",
              "      <td>-0.048526</td>\n",
              "      <td>0.003726</td>\n",
              "      <td>-0.046989</td>\n",
              "      <td>-0.072583</td>\n",
              "      <td>-0.229890</td>\n",
              "      <td>-0.051829</td>\n",
              "      <td>-0.186431</td>\n",
              "      <td>0.005577</td>\n",
              "      <td>-0.227811</td>\n",
              "      <td>-0.093958</td>\n",
              "      <td>0.157381</td>\n",
              "      <td>0.199125</td>\n",
              "      <td>0.103656</td>\n",
              "      <td>-0.112644</td>\n",
              "      <td>0.018395</td>\n",
              "      <td>0.134007</td>\n",
              "      <td>0.107792</td>\n",
              "      <td>-0.257777</td>\n",
              "      <td>0.151081</td>\n",
              "      <td>-0.152100</td>\n",
              "      <td>0.164346</td>\n",
              "      <td>0.258414</td>\n",
              "      <td>-0.012498</td>\n",
              "      <td>0.061118</td>\n",
              "      <td>-0.179063</td>\n",
              "      <td>-0.186134</td>\n",
              "      <td>0.221206</td>\n",
              "      <td>0.333459</td>\n",
              "      <td>-0.233890</td>\n",
              "      <td>0.142896</td>\n",
              "      <td>-0.064313</td>\n",
              "      <td>0.088227</td>\n",
              "      <td>-0.296398</td>\n",
              "      <td>0.056088</td>\n",
              "      <td>0.079036</td>\n",
              "      <td>0.042217</td>\n",
              "      <td>0.102526</td>\n",
              "      <td>-0.221315</td>\n",
              "      <td>-0.108310</td>\n",
              "      <td>-0.164680</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>WI</th>\n",
              "      <td>0.085855</td>\n",
              "      <td>-0.026329</td>\n",
              "      <td>0.188422</td>\n",
              "      <td>0.208438</td>\n",
              "      <td>0.272377</td>\n",
              "      <td>-0.073486</td>\n",
              "      <td>0.254118</td>\n",
              "      <td>0.188227</td>\n",
              "      <td>0.189222</td>\n",
              "      <td>0.166843</td>\n",
              "      <td>-0.016172</td>\n",
              "      <td>0.068667</td>\n",
              "      <td>-0.021109</td>\n",
              "      <td>0.048562</td>\n",
              "      <td>0.160558</td>\n",
              "      <td>0.095011</td>\n",
              "      <td>0.158244</td>\n",
              "      <td>0.168777</td>\n",
              "      <td>0.326352</td>\n",
              "      <td>0.070765</td>\n",
              "      <td>0.236266</td>\n",
              "      <td>-0.173682</td>\n",
              "      <td>0.059607</td>\n",
              "      <td>-0.020287</td>\n",
              "      <td>-0.088031</td>\n",
              "      <td>-0.083667</td>\n",
              "      <td>-0.222770</td>\n",
              "      <td>-0.197498</td>\n",
              "      <td>-0.101200</td>\n",
              "      <td>0.085178</td>\n",
              "      <td>-0.205141</td>\n",
              "      <td>-0.099795</td>\n",
              "      <td>0.141622</td>\n",
              "      <td>-0.227501</td>\n",
              "      <td>-0.160229</td>\n",
              "      <td>-0.205250</td>\n",
              "      <td>0.046487</td>\n",
              "      <td>0.017665</td>\n",
              "      <td>0.237532</td>\n",
              "      <td>-0.027263</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>AFG</th>\n",
              "      <td>0.313823</td>\n",
              "      <td>-0.050491</td>\n",
              "      <td>0.089990</td>\n",
              "      <td>-0.001992</td>\n",
              "      <td>0.204509</td>\n",
              "      <td>-0.035358</td>\n",
              "      <td>0.072658</td>\n",
              "      <td>-0.049220</td>\n",
              "      <td>0.147889</td>\n",
              "      <td>-0.028405</td>\n",
              "      <td>0.077874</td>\n",
              "      <td>-0.158315</td>\n",
              "      <td>-0.061597</td>\n",
              "      <td>-0.341433</td>\n",
              "      <td>0.341536</td>\n",
              "      <td>-0.136989</td>\n",
              "      <td>-0.244974</td>\n",
              "      <td>0.001855</td>\n",
              "      <td>0.004089</td>\n",
              "      <td>-0.207143</td>\n",
              "      <td>0.009309</td>\n",
              "      <td>0.125648</td>\n",
              "      <td>0.285290</td>\n",
              "      <td>-0.078250</td>\n",
              "      <td>0.002577</td>\n",
              "      <td>0.171165</td>\n",
              "      <td>-0.121502</td>\n",
              "      <td>-0.016939</td>\n",
              "      <td>-0.139395</td>\n",
              "      <td>0.202336</td>\n",
              "      <td>0.046296</td>\n",
              "      <td>0.064708</td>\n",
              "      <td>-0.080982</td>\n",
              "      <td>0.131887</td>\n",
              "      <td>-0.190532</td>\n",
              "      <td>-0.115175</td>\n",
              "      <td>-0.282121</td>\n",
              "      <td>0.089471</td>\n",
              "      <td>0.173993</td>\n",
              "      <td>0.147389</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>WC</th>\n",
              "      <td>-0.223854</td>\n",
              "      <td>0.097253</td>\n",
              "      <td>0.275048</td>\n",
              "      <td>-0.156086</td>\n",
              "      <td>-0.053700</td>\n",
              "      <td>-0.232419</td>\n",
              "      <td>-0.079267</td>\n",
              "      <td>0.338941</td>\n",
              "      <td>-0.037217</td>\n",
              "      <td>-0.075062</td>\n",
              "      <td>0.385352</td>\n",
              "      <td>0.250885</td>\n",
              "      <td>-0.105916</td>\n",
              "      <td>-0.267116</td>\n",
              "      <td>0.005446</td>\n",
              "      <td>0.167402</td>\n",
              "      <td>0.116006</td>\n",
              "      <td>-0.286505</td>\n",
              "      <td>0.339166</td>\n",
              "      <td>0.004812</td>\n",
              "      <td>-0.007682</td>\n",
              "      <td>0.135986</td>\n",
              "      <td>-0.082327</td>\n",
              "      <td>0.058803</td>\n",
              "      <td>-0.033615</td>\n",
              "      <td>-0.027307</td>\n",
              "      <td>0.041902</td>\n",
              "      <td>0.001863</td>\n",
              "      <td>-0.105001</td>\n",
              "      <td>-0.047812</td>\n",
              "      <td>-0.019827</td>\n",
              "      <td>0.101611</td>\n",
              "      <td>-0.138606</td>\n",
              "      <td>-0.064110</td>\n",
              "      <td>-0.016651</td>\n",
              "      <td>0.019896</td>\n",
              "      <td>-0.120674</td>\n",
              "      <td>-0.048749</td>\n",
              "      <td>0.142771</td>\n",
              "      <td>-0.070017</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>201 rows × 40 columns</p>\n",
              "</div>"
            ],
            "text/plain": [
              "                      0         1         2   ...        37        38        39\n",
              "nodes                                         ...                              \n",
              "Kedar Jadhav    0.292280  0.056647 -0.262089  ... -0.278417 -0.054648  0.082107\n",
              "Mayank Agarwal  0.272155  0.099263 -0.243889  ... -0.286260 -0.052138  0.114731\n",
              "Rohit Sharma    0.316569  0.044673 -0.267582  ... -0.271688 -0.040246  0.106200\n",
              "Shikhar Dhawan  0.269974  0.045036 -0.268864  ... -0.288133 -0.051525  0.107410\n",
              "Virat Kohli     0.305277  0.046292 -0.283788  ... -0.265505 -0.044504  0.125493\n",
              "...                  ...       ...       ...  ...       ...       ...       ...\n",
              "SA             -0.084763 -0.113558 -0.083624  ...  0.147233 -0.172231 -0.327650\n",
              "BAN            -0.048526  0.003726 -0.046989  ... -0.221315 -0.108310 -0.164680\n",
              "WI              0.085855 -0.026329  0.188422  ...  0.017665  0.237532 -0.027263\n",
              "AFG             0.313823 -0.050491  0.089990  ...  0.089471  0.173993  0.147389\n",
              "WC             -0.223854  0.097253  0.275048  ... -0.048749  0.142771 -0.070017\n",
              "\n",
              "[201 rows x 40 columns]"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 89
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "cX0gCKkpbSJ5",
        "outputId": "dc03bb50-f6eb-4976-f138-9d80b361b1c0",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "source": [
        "from sklearn.metrics.pairwise import cosine_similarity\n",
        "a = 'Virat Kohli'\n",
        "b = ['Mayank Agarwal','Yuzvendra Chahal','David Warner','Soumya Sarkar','Faf du Plessis']\n",
        "\n",
        "for each in b:\n",
        "  x=node_embedd.loc[[a]]\n",
        "  y=node_embedd.loc[[each]]\n",
        "  print(abs(cosine_similarity(x,y)),a,\"<----->\" ,each)"
      ],
      "execution_count": 92,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "[[0.98712146]] Virat Kohli <-----> Mayank Agarwal\n",
            "[[0.81957287]] Virat Kohli <-----> Yuzvendra Chahal\n",
            "[[0.32843906]] Virat Kohli <-----> David Warner\n",
            "[[0.21252364]] Virat Kohli <-----> Soumya Sarkar\n",
            "[[0.37364566]] Virat Kohli <-----> Faf du Plessis\n"
          ],
          "name": "stdout"
        }
      ]
    }
  ]
}