draw_loss_figure_noAllocation.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import random\n",
    "from libs.node import Node\n",
    "from libs.node import Station\n",
    "from libs.node import StationDcf\n",
    "from libs.node import StationRl\n",
    "from libs.channel import Channel\n",
    "from tqdm import tqdm\n",
    "from config import Config\n",
    "import matplotlib.pyplot as plt\n",
    "import pylab as pl\n",
    "from mpl_toolkits.axes_grid1 import host_subplot"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cfg = Config()\n",
    "global_time = 0\n",
    "channel = Channel(global_time, [])\n",
    "\n",
    "station_num = 5\n",
    "data_rate = 6  # Mbps\n",
    "# All the lengeth is a mutible of slot\n",
    "\n",
    "# incule header\n",
    "pkt_len = 1560\n",
    "# us\n",
    "slot_len = 10\n",
    "sifs = 2\n",
    "ack_len = 2 + sifs\n",
    "difs = 4\n",
    "timeout = ack_len\n",
    "frame_len = pkt_len * 8 / slot_len / data_rate"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def draw_loss_figure(stations_list, NUM_EPOCHS):\n",
    "    for i in range(len(stations_list)):\n",
    "        station = stations_list[i]\n",
    "        loss = station.model.lossHitory\n",
    "        # print(len(loss))\n",
    "\n",
    "        host = host_subplot(111)  # row=1 col=1 first pic\n",
    "        # ajust the right boundary of the plot window\n",
    "        plt.subplots_adjust(right=0.8)\n",
    "        par1 = host.twinx()   # 共享x轴\n",
    "\n",
    "        # set labels\n",
    "        host.set_xlabel(\"steps\")\n",
    "        host.set_ylabel(\"loss\")\n",
    "        # par1.set_ylabel(\"test-accuracy\")\n",
    "\n",
    "        # plot curves\n",
    "        # grid = range(min(len(loss), len(loss_2), len(loss_3)))\n",
    "        p1, = host.plot(range(len(loss)), loss, label=\"loss\")\n",
    "        # p2, = host.plot(range(len(loss_2)), loss_2, label=\"loss 2\")\n",
    "\n",
    "        # set location of the legend,\n",
    "        # 1->rightup corner, 2->leftup corner, 3->leftdown corner\n",
    "        # 4->rightdown corner, 5->rightmid ...\n",
    "        host.legend(loc=5)\n",
    "\n",
    "        # set label color\n",
    "        host.axis[\"left\"].label.set_color(p1.get_color())\n",
    "        # host.axis[\"left\"].label.set_color(p2.get_color())\n",
    "        # host.axis[\"left\"].label.set_color(p3.get_color())\n",
    "        # par1.axis[\"right\"].label.set_color(p2.get_color())\n",
    "\n",
    "        # set the range of x axis of host and y axis of par1\n",
    "        # host.set_xlim([-200, 5200])\n",
    "        # par1.set_ylim([-0.1, 1.1])\n",
    "\n",
    "        plt.draw()\n",
    "        \n",
    "        plt.savefig('./fig/noAllocation/Epoch_{}_station_{}.jpg'.format(NUM_EPOCHS, i))\n",
    "        \n",
    "        plt.show()\n",
    "        plt.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''\n",
    "NUM_EPOCHS = 1000\n",
    "'''\n",
    "cfg.NUM_EPOCHS = 1000\n",
    "stations_list_1000 = []\n",
    "total_time = 0\n",
    "for i in range(station_num):\n",
    "    if cfg.stationType == \"Dcf\":\n",
    "        station = StationDcf(i, frame_len, channel,\n",
    "                             global_time, i, timeout, ack_len, difs, sifs)\n",
    "    elif cfg.stationType == \"RL\":\n",
    "        station = StationRl(i, frame_len, channel,\n",
    "                            global_time, i, timeout, ack_len, (i+1))\n",
    "    stations_list_1000.append(station)\n",
    "\n",
    "if cfg.stationType == \"Dcf\":\n",
    "    startEpoch = 0\n",
    "elif cfg.stationType == \"RL\":\n",
    "    startEpoch = stations_list_1000[0].epoch\n",
    "\n",
    "print(\"==> startEpoch: \", startEpoch)\n",
    "for i in tqdm(range(startEpoch, startEpoch+cfg.NUM_EPOCHS)):\n",
    "    for station in stations_list_1000:\n",
    "        station.simulate(global_time)\n",
    "    global_time = global_time + 1\n",
    "    channel.update_state(global_time)\n",
    "\n",
    "    if cfg.shuffleStationList:\n",
    "        random.shuffle(stations_list_1000)\n",
    "\n",
    "for station in stations_list_1000:\n",
    "    total_time += station.total_pkt_time\n",
    "\n",
    "    if cfg.stationType == \"RL\" and cfg.saveModel:\n",
    "        station.saveModel()\n",
    "print(\"==> total_time:\", total_time)\n",
    "total_time_channel = 0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "draw_loss_figure(stations_list_1000, 1000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''\n",
    "NUM_EPOCHS = 5000\n",
    "'''\n",
    "cfg.NUM_EPOCHS = 5000\n",
    "stations_list_5000 = []\n",
    "total_time = 0\n",
    "for i in range(station_num):\n",
    "    if cfg.stationType == \"Dcf\":\n",
    "        station = StationDcf(i, frame_len, channel,\n",
    "                             global_time, i, timeout, ack_len, difs, sifs)\n",
    "    elif cfg.stationType == \"RL\":\n",
    "        station = StationRl(i, frame_len, channel,\n",
    "                            global_time, i, timeout, ack_len, (i+1))\n",
    "    stations_list_5000.append(station)\n",
    "\n",
    "if cfg.stationType == \"Dcf\":\n",
    "    startEpoch = 0\n",
    "elif cfg.stationType == \"RL\":\n",
    "    startEpoch = stations_list_5000[0].epoch\n",
    "\n",
    "print(\"==> startEpoch: \", startEpoch)\n",
    "for i in tqdm(range(startEpoch, startEpoch+cfg.NUM_EPOCHS)):\n",
    "    for station in stations_list_5000:\n",
    "        station.simulate(global_time)\n",
    "    global_time = global_time + 1\n",
    "    channel.update_state(global_time)\n",
    "\n",
    "    if cfg.shuffleStationList:\n",
    "        random.shuffle(stations_list_5000)\n",
    "\n",
    "for station in stations_list_5000:\n",
    "    total_time += station.total_pkt_time\n",
    "\n",
    "    if cfg.stationType == \"RL\" and cfg.saveModel:\n",
    "        station.saveModel()\n",
    "print(\"==> total_time:\", total_time)\n",
    "total_time_channel = 0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "draw_loss_figure(stations_list_5000, 5000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''\n",
    "NUM_EPOCHS = 10000\n",
    "'''\n",
    "cfg.NUM_EPOCHS = 10000\n",
    "stations_list_10000 = []\n",
    "total_time = 0\n",
    "for i in range(station_num):\n",
    "    if cfg.stationType == \"Dcf\":\n",
    "        station = StationDcf(i, frame_len, channel,\n",
    "                             global_time, i, timeout, ack_len, difs, sifs)\n",
    "    elif cfg.stationType == \"RL\":\n",
    "        station = StationRl(i, frame_len, channel,\n",
    "                            global_time, i, timeout, ack_len, (i+1))\n",
    "    stations_list_10000.append(station)\n",
    "\n",
    "if cfg.stationType == \"Dcf\":\n",
    "    startEpoch = 0\n",
    "elif cfg.stationType == \"RL\":\n",
    "    startEpoch = stations_list_10000[0].epoch\n",
    "\n",
    "print(\"==> startEpoch: \", startEpoch)\n",
    "for i in tqdm(range(startEpoch, startEpoch+cfg.NUM_EPOCHS)):\n",
    "    for station in stations_list_10000:\n",
    "        station.simulate(global_time)\n",
    "    global_time = global_time + 1\n",
    "    channel.update_state(global_time)\n",
    "\n",
    "    if cfg.shuffleStationList:\n",
    "        random.shuffle(stations_list_10000)\n",
    "\n",
    "# for station in stations_list_10000:\n",
    "#     total_time += station.total_pkt_time\n",
    "\n",
    "#     if cfg.stationType == \"RL\" and cfg.saveModel:\n",
    "#         station.saveModel()\n",
    "        \n",
    "# print(\"==> total_time:\", total_time)\n",
    "# total_time_channel = 0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "draw_loss_figure(stations_list_10000, 10000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''\n",
    "NUM_EPOCHS = 20000\n",
    "'''\n",
    "cfg.NUM_EPOCHS = 20000\n",
    "stations_list_20000 = []\n",
    "total_time = 0\n",
    "for i in range(station_num):\n",
    "    if cfg.stationType == \"Dcf\":\n",
    "        station = StationDcf(i, frame_len, channel,\n",
    "                             global_time, i, timeout, ack_len, difs, sifs)\n",
    "    elif cfg.stationType == \"RL\":\n",
    "        station = StationRl(i, frame_len, channel,\n",
    "                            global_time, i, timeout, ack_len, (i+1))\n",
    "    stations_list_20000.append(station)\n",
    "\n",
    "if cfg.stationType == \"Dcf\":\n",
    "    startEpoch = 0\n",
    "elif cfg.stationType == \"RL\":\n",
    "    startEpoch = stations_list_20000[0].epoch\n",
    "\n",
    "print(\"==> startEpoch: \", startEpoch)\n",
    "for i in tqdm(range(startEpoch, startEpoch+cfg.NUM_EPOCHS)):\n",
    "    for station in stations_list_20000:\n",
    "        station.simulate(global_time)\n",
    "    global_time = global_time + 1\n",
    "    channel.update_state(global_time)\n",
    "\n",
    "    if cfg.shuffleStationList:\n",
    "        random.shuffle(stations_list_20000)\n",
    "\n",
    "# for station in stations_list_20000:\n",
    "#     total_time += station.total_pkt_time\n",
    "\n",
    "#     if cfg.stationType == \"RL\" and cfg.saveModel:\n",
    "#         station.saveModel()\n",
    "# print(\"==> total_time:\", total_time)\n",
    "# total_time_channel = 0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "draw_loss_figure(stations_list_20000, 20000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.7.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}