mainbpr.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "4f81dbcc",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Reading files...\n",
      "Finished Reading files.\n",
      "Iteration 0\n",
      "Iteration 1\n",
      "Iteration 2\n",
      "Iteration 3\n",
      "Iteration 4\n",
      "Iteration 5\n",
      "Iteration 6\n",
      "Iteration 7\n",
      "Iteration 8\n",
      "Iteration 9\n",
      "Finished updating.\n",
      "Calculating Hit Rate...\n",
      "Trained:\n",
      "HitRate 0.5073891625615764\n",
      "Number of Users:609\n",
      "Number of Movies:9742\n"
     ]
    }
   ],
   "source": [
    "from filereader import read_ratings, read_movies\n",
    "from bpr import bpr_update\n",
    "from hitrate import hit_rate\n",
    "from random import seed\n",
    "from util import cov_matrix\n",
    "from util import random_vector\n",
    "import numpy as np\n",
    "import random as random\n",
    "seed(42)\n",
    "np.random.seed(42)\n",
    "print(\"Reading files...\")\n",
    "users = read_ratings(\"data/ratings.csv\")\n",
    "movies = read_movies(\"data/movies.csv\")\n",
    "print(\"Finished Reading files.\")\n",
    "iterations = 10\n",
    "\n",
    "for i in range(iterations):\n",
    "    print(\"Iteration \" + str(i))\n",
    "    bpr_update(users, movies)\n",
    "\n",
    "print(\"Finished updating.\")\n",
    "print(\"Calculating Hit Rate...\")\n",
    "\n",
    "hits, denom, rms, rmsall = hit_rate(users, movies)\n",
    "\n",
    "print(\"Trained:\")\n",
    "print(\"HitRate\" + ' ' + str(hits/denom))\n",
    "print(\"Number of Users:\" + str(len(users)))\n",
    "print(\"Number of Movies:\" + str(len(movies)))"
   ]
  }
 ],
 "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.8.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}