From 160c43046c70ce5002cba1edaee73ee46648fd91 Mon Sep 17 00:00:00 2001
From: tarzan <1334512682@qq.com>
Date: Thu, 13 May 2021 11:45:32 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8E=A8=E8=8D=90=E7=AE=97=E6=B3=95=E4=BF=AE?=
 =?UTF-8?q?=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pom.xml                                               |  5 -----
 .../tarzan/recommend/RecommendSystemApplication.java  |  9 ++++++++-
 .../com/tarzan/recommend/Service/FileDataSource.java  |  3 ++-
 .../java/com/tarzan/recommend/Service/Recommend.java  | 11 +++--------
 4 files changed, 13 insertions(+), 15 deletions(-)

diff --git a/pom.xml b/pom.xml
index dab3d29..b78f4d8 100644
--- a/pom.xml
+++ b/pom.xml
@@ -13,17 +13,14 @@
 	<version>0.0.1-SNAPSHOT</version>
 	<name>recommend_system</name>
 	<description>Demo project for Spring Boot</description>
-
 	<properties>
 		<java.version>1.8</java.version>
 	</properties>
-
 	<dependencies>
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter</artifactId>
 		</dependency>
-
 		<dependency>
 			<groupId>org.projectlombok</groupId>
 			<artifactId>lombok</artifactId>
@@ -47,7 +44,6 @@
 			<scope>compile</scope>
 		</dependency>
 	</dependencies>
-
 	<build>
 		<plugins>
 			<plugin>
@@ -56,5 +52,4 @@
 			</plugin>
 		</plugins>
 	</build>
-
 </project>
diff --git a/src/main/java/com/tarzan/recommend/RecommendSystemApplication.java b/src/main/java/com/tarzan/recommend/RecommendSystemApplication.java
index 89b61c3..01dc0f9 100644
--- a/src/main/java/com/tarzan/recommend/RecommendSystemApplication.java
+++ b/src/main/java/com/tarzan/recommend/RecommendSystemApplication.java
@@ -1,13 +1,20 @@
 package com.tarzan.recommend;
 
+import com.tarzan.recommend.Service.Recommend;
+import com.tarzan.recommend.dto.ItemDTO;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 
+import java.util.List;
+
 @SpringBootApplication
 public class RecommendSystemApplication {
 
 	public static void main(String[] args) {
-		SpringApplication.run(RecommendSystemApplication.class, args);
+		//SpringApplication.run(RecommendSystemApplication.class, args);
+		List<ItemDTO> itemList= Recommend.guessUserLike(2);
+		System.out.println("------猜你可能喜欢---------------下列电影");
+		itemList.forEach(e-> System.out.println(e.getName()));
 	}
 
 }
diff --git a/src/main/java/com/tarzan/recommend/Service/FileDataSource.java b/src/main/java/com/tarzan/recommend/Service/FileDataSource.java
index ad09a11..7009ce6 100644
--- a/src/main/java/com/tarzan/recommend/Service/FileDataSource.java
+++ b/src/main/java/com/tarzan/recommend/Service/FileDataSource.java
@@ -17,7 +17,7 @@ import java.util.List;
 public class FileDataSource {
 
 
-      public final static String folderPath="F:\\ml-100k";
+      public  static String folderPath;
 
 
     /**
@@ -30,6 +30,7 @@ public class FileDataSource {
      * @date 2020年07月31日 16:53:40
      */
     public static List<RelateDTO> getData() {
+        folderPath=new FileDataSource().getClass().getResource("/ml-100k").getPath();
         List<RelateDTO> relateList = Lists.newArrayList();
         try {
             FileInputStream out = new FileInputStream(folderPath+"\\u.data");
diff --git a/src/main/java/com/tarzan/recommend/Service/Recommend.java b/src/main/java/com/tarzan/recommend/Service/Recommend.java
index 044f409..3dee032 100644
--- a/src/main/java/com/tarzan/recommend/Service/Recommend.java
+++ b/src/main/java/com/tarzan/recommend/Service/Recommend.java
@@ -18,27 +18,22 @@ import java.util.stream.Collectors;
  */
 public class Recommend{
 
-
     /**
      * 方法描述: 猜你喜欢
      *
      * @param
-     * @Return {@link List< ItemDTO>}
+     * @Return {@link List<ItemDTO>}
      * @throws
      * @author tarzan
      * @date 2020年07月31日 17:28:06
      */
-    public static List<ItemDTO>  guessYouLike(){
+    public static List<ItemDTO>  guessUserLike(int userId){
         CoreMath coreMath = new CoreMath();
         List<RelateDTO> data= FileDataSource.getData();
-        List<Integer> recommendations = coreMath.recommend(2, data);
+        List<Integer> recommendations = coreMath.recommend(userId, data);
         List<ItemDTO> itemList= FileDataSource.getItemData().stream().filter(e->recommendations.contains(e.getId())).collect(Collectors.toList());
         return itemList;
     }
 
-    public static void main(String[] args) {
-        List<ItemDTO> itemList= Recommend.guessYouLike();
-        System.out.println("------猜你可能喜欢---------------下列电影="+itemList.stream().map(e->e.getName()).collect(Collectors.toList()));
-    }
 
 }