-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmain.js
56 lines (48 loc) · 1.6 KB
/
main.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
const data = require("./data/data.js");
// Algorithms
const Euclidean = require("./algorithm/euclidean.js");
const Pearson = require("./algorithm/pearson.js");
const Match = require("./core/match.js");
const UserBased = require("./core/user-based.js");
const ItemBased = require("./core/item-based.js");
const Similarity = require("./core/similarity.js");
const person1 = "Toby";
const person2 = "William";
const resultsWithEuclidean = Similarity(
data,
person1,
person2,
Euclidean.similarity
);
const resultsWithPearson = Similarity(
data,
person1,
person2,
Pearson.similarity
);
console.log(
`Similarity between ${person1} and ${person2} with Euclidean: ${resultsWithEuclidean.toFixed(
5
)}`
);
console.log("\n");
console.log(
`Similarity between ${person1} and ${person2} with Pearson: ${resultsWithPearson.toFixed(
5
)}`
);
console.log("\n");
const matchesWithEuclidean = Match(data, person1, 3, Euclidean.similarity);
console.log(`Top matches for ${person1} with Euclidean: `);
matchesWithEuclidean.forEach((p) => console.log(p.user, p.score));
console.log("\n");
const matchesWithPearson = Match(data, person1, 3, Pearson.similarity);
console.log(`Top matches for ${person1} with Pearson: `);
matchesWithPearson.forEach((p) => console.log(p.user, p.score));
console.log("\n");
// user based collaborative filtering
const recommendation = UserBased(data, person1, Euclidean.similarity);
console.log(`Recommendations for ${person1}`);
recommendation.forEach((r) => console.log(r.user, r.score));
console.log("\n");
console.log(ItemBased.recommendations(data, Euclidean.similarity, "Toby"));