Created
February 10, 2019 16:56
-
-
Save Kamilahsantos/543ce9f1319e3b9a62f6d739de03465d to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package br.com.kamila; | |
import org.apache.mahout.cf.taste.impl.model.file.FileDataModel; | |
import org.apache.mahout.cf.taste.model.DataModel; | |
import java.io.File; | |
import java.io.IOException; | |
//nesta classe está a relação dos modelos de filmes e livros para a serem recomendados | |
public class Recomendador { | |
public DataModel getModeloLivros() throws IOException { | |
//arquivo usado com a base de usuários, os códigos dos livros e as notas atribuídas | |
return getModelo("livros.csv"); | |
} | |
private DataModel getModelo(String path) throws IOException { | |
File file = new File("src/main/resources/" + path); | |
return new FileDataModel(file); | |
} | |
public DataModel getModeloDeFilmes() throws IOException { | |
//arquivo usado com a base de usuários, os códigos dos filmes e as notas atribuídas | |
return getModelo("filmes.csv"); | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment