LFM2.ipynb · czkn/travel-Recommended-System

加入 Gitee

与超过 1200万开发者一起发现、参与优秀开源项目，私有仓库也完全免费：）

文件

该仓库未声明开源许可证文件（LICENSE），使用请关注具体项目描述及其代码上游依赖。

克隆/下载

LFM2.ipynb 2.82 KB

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "#coding = utf-8\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "import math \n",
    "import random\n",
    "import pickle\n",
    "import os\n",
    "import csv\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "result = r'C:\\Users\\Evilemon\\Desktop\\recommender\\exam\\result'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "def Recommendation(user_ID, K, result):\n",
    "    \n",
    "    user_id = user_ID.join('.0')\n",
    "    print(user_id)\n",
    "    jd_dict_path = os.path.join(result, 'jd_dict')\n",
    "    user_dict_path = os.path.join(result, 'user_dict')\n",
    "    user_factor_matrix_path = os.path.join(result, 'user_factor_matrix.npy')\n",
    "    jd_factor_matrix_path = os.path.join(result, 'jd_factor_matrix.npy')\n",
    "    \n",
    "    jd_dict = np.load(jd_dict_path)\n",
    "    user_dict = np.load(user_dict_path)\n",
    "    P = np.load(user_factor_matrix_path)\n",
    "    Q = np.load(jd_factor_matrix_path)\n",
    "\n",
    "    user = random.randint(0,1000)\n",
    "    for k, v in user_dict.items():\n",
    "        if str(k) == user_id:\n",
    "            user = v\n",
    "        \n",
    "    rank = dict()\n",
    "    for i in range(2541):\n",
    "        rank[i] = (np.multiply(P[:,user], Q[:,i])).sum()\n",
    "    a = zip(rank.values(), rank.keys())\n",
    "    b = list(sorted(a, reverse = True))   \n",
    "    c = b[0:K]\n",
    "    rank = list()\n",
    "    for m in range(K):\n",
    "        jn = c[m][1]\n",
    "        for k, v in jd_dict.items():\n",
    "            if v == jn:\n",
    "                K = k[:-2]\n",
    "                rank.append(K)\n",
    "    return rank"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      ".1r0\n",
      "['704127', '710545', '702719', '706190', '713994', '714791', '715470', '7475306', '5257837', '711995', '708991', '710603', '704630', '715478', '716377', '722948', '713871', '713204', '710170', '704600']\n"
     ]
    }
   ],
   "source": [
    "rank = Recommendation('1r', 20, result)\n",
    "print(rank)"
   ]
  }
 ],
 "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.6.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}

一键复制编辑原始数据按行查看历史

提交于 2019-07-06 13:37 . Add files via upload

马建仓 AI 助手

尝试更多

代码解读

代码找茬

代码优化

https://gitee.com/czkn/travel-recommended-system.git

git@gitee.com:czkn/travel-recommended-system.git

czkn

travel-recommended-system

travel-Recommended-System

master

czkn/travel-Recommended-System

简介

发行版

贡献者

近期动态

czkn/travel-Recommended-System .gitee-modal { width: 500px !important; }

简介

发行版

贡献者

近期动态

搜索帮助

czkn/travel-Recommended-System