lennon/recommend_system
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

推荐算法修改

Browse Source
This commit is contained in:
tarzan
2021-05-13 13:36:27 +08:00
parent 160c43046c
commit c2d7dd4647
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/tarzan/recommend/Service/FileDataSource.java
View File

@@ -6,7 +6,6 @@ import com.tarzan.recommend.dto.UserDTO;
import lombok.Data; import lombok.Data;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.assertj.core.util.Lists; import org.assertj.core.util.Lists;
import org.springframework.util.ResourceUtils;
import java.io.*; import java.io.*;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
@@ -62,6 +61,7 @@ public class FileDataSource {
* @date 2020年07月31日 16:54:51 * @date 2020年07月31日 16:54:51
*/ */
public static List<UserDTO> getUserData() { public static List<UserDTO> getUserData() {
folderPath=new FileDataSource().getClass().getResource("/ml-100k").getPath();
List<UserDTO> userList = Lists.newArrayList(); List<UserDTO> userList = Lists.newArrayList();
try { try {
FileInputStream out = new FileInputStream(folderPath+"\\u.user"); FileInputStream out = new FileInputStream(folderPath+"\\u.user");
@@ -96,6 +96,7 @@ public class FileDataSource {
* @date 2020年07月31日 16:54:22 * @date 2020年07月31日 16:54:22
*/ */
public static List<ItemDTO> getItemData() { public static List<ItemDTO> getItemData() {
folderPath=new FileDataSource().getClass().getResource("/ml-100k").getPath();
List<ItemDTO> itemList = Lists.newArrayList(); List<ItemDTO> itemList = Lists.newArrayList();
try { try {
FileInputStream out = new FileInputStream(folderPath+"\\u.item"); FileInputStream out = new FileInputStream(folderPath+"\\u.item");