Skip to content

Instantly share code, notes, and snippets.

@anupamchugh
Last active November 15, 2023 10:48
Show Gist options
  • Save anupamchugh/bfe28a774668259c249a4788f3688152 to your computer and use it in GitHub Desktop.
Save anupamchugh/bfe28a774668259c249a4788f3688152 to your computer and use it in GitHub Desktop.
public class Recommender: ObservableObject {
@Published var movies = [Movie]()
init(){
load()
}
func load() {
do{
let recommender = MovieRecommender()
let input = MovieRecommenderInput(items: [:], k: 10, restrict_: [], exclude: [])
let result = try recommender.prediction(input: input)
var tempMovies = [Movie]()
for str in result.recommendations{
let score = result.scores[str] ?? 0
tempMovies.append(Movie(name: "\(str)", score: score))
}
self.movies = tempMovies
}catch(let error){
print("error is \(error.localizedDescription)")
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment