Skip to content

Commit ee865de

Browse files
committed
Updated overrides based on suggestions
1 parent 732f3ca commit ee865de

File tree

2 files changed

+85
-37
lines changed

2 files changed

+85
-37
lines changed

name_server/data_processing/aggregate_name_suggestions.js

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ const runAggregation = async (aggregationPipeline, collection) => {
7171

7272
const coll = client.db("river_runner").collection(collection);
7373

74-
coll.aggregate(aggregationPipeline, async (cmdErr, result) => {
74+
coll.aggregate(aggregationPipeline, { allowDiskUse: true, cursor: { batchSize: 1000 } }, async (cmdErr, result) => {
7575
assert.equal(null, cmdErr);
7676

7777
const aggregationData = await result.toArray();
@@ -88,43 +88,43 @@ const main = async () => {
8888
const csvFilepath = "../../public/data/name_overrides.csv";
8989

9090
const existingOverrides = await getExistingOverrides(csvFilepath);
91-
const groupedOccurences = await runAggregation(unnamedFeatureCounts, "unnamed_features");
91+
// const groupedOccurences = await runAggregation(unnamedFeatureCounts, "unnamed_features");
9292
const groupedSuggestions = await runAggregation(suggestionCounts, "suggestions");
9393

94-
// const overridenIDs = existingOverrides.map((d) => Number(d.levelpathid));
95-
// const unhandledSuggestions = groupedSuggestions
96-
// .filter((d) => d.count > 3)
97-
// .filter((d) => !overridenIDs.includes(d._id));
98-
99-
// console.log(unhandledSuggestions);
100-
const joinedData = groupedOccurences.map((d) => {
101-
const suggestions = groupedSuggestions.find((a) => a._id === d._id) || {
102-
count: 0,
103-
route_url: [],
104-
suggestions: null,
105-
};
106-
107-
const override = existingOverrides.find((a) => Number(a.levelpathid) === d._id);
108-
109-
return {
110-
levelpathid: d._id,
111-
sample_route_url: [...suggestions.route_url, d.route_url].filter((d) => d)[0],
112-
num_suggestions: suggestions.count,
113-
num_occurences: d.count,
114-
suggestions: suggestions.suggestions,
115-
overriden: override ? true : false,
116-
overriden_val: override ? override.feature_name : null,
117-
};
118-
});
119-
120-
const outputFile = fs.createWriteStream("data/aggregated_suggestions.csv");
121-
122-
csv
123-
.write(joinedData, { headers: true })
124-
.on("finish", function () {
125-
console.log("Write to CSV successfully!");
126-
})
127-
.pipe(outputFile);
94+
const overridenIDs = existingOverrides.map((d) => Number(d.levelpathid));
95+
const unhandledSuggestions = groupedSuggestions
96+
.filter((d) => d.count > 3)
97+
.filter((d) => !overridenIDs.includes(d._id));
98+
99+
console.log(unhandledSuggestions);
100+
// const joinedData = groupedOccurences.map((d) => {
101+
// const suggestions = groupedSuggestions.find((a) => a._id === d._id) || {
102+
// count: 0,
103+
// route_url: [],
104+
// suggestions: null,
105+
// };
106+
107+
// const override = existingOverrides.find((a) => Number(a.levelpathid) === d._id);
108+
109+
// return {
110+
// levelpathid: d._id,
111+
// sample_route_url: [...suggestions.route_url, d.route_url].filter((d) => d)[0],
112+
// num_suggestions: suggestions.count,
113+
// num_occurences: d.count,
114+
// suggestions: suggestions.suggestions,
115+
// overriden: override ? true : false,
116+
// overriden_val: override ? override.feature_name : null,
117+
// };
118+
// });
119+
120+
// const outputFile = fs.createWriteStream("data/aggregated_suggestions.csv");
121+
122+
// csv
123+
// .write(joinedData, { headers: true })
124+
// .on("finish", function () {
125+
// console.log("Write to CSV successfully!");
126+
// })
127+
// .pipe(outputFile);
128128
};
129129

130130
main();

public/data/name_overrides.csv

Lines changed: 49 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,4 +220,52 @@ levelpathid,feature_name
220220
1499364,Soar
221221
775504,Verde
222222
835801,Rosebud
223-
834198,Red Deer
223+
834198,Red Deer
224+
1496234,Berounka
225+
1504880,Töss
226+
1504556,Emme
227+
1504398,Nidda
228+
1416325,Drac
229+
1504319,Sieg
230+
1522090,Mondego
231+
1516793,Hers-Mort
232+
1516780,Aveyron
233+
1469298,Iller
234+
1416435,Ubaye
235+
1500212,Fulda
236+
1423150,Valira
237+
1496257,Sázava
238+
1504595,Amblève
239+
1467245,Danube
240+
863697,Thames
241+
1505211,Suhre
242+
1504371,Erft
243+
1416292,Ardèche
244+
1513466,Clain
245+
1505530,Lorze
246+
1468237,Iskar
247+
1508236,Wye
248+
1087803,Krokodil
249+
1468691,Lech
250+
1496836,Alster
251+
1468306,Salzach
252+
1562047,Eider
253+
1522591,Manzanares
254+
1494757,Tweed
255+
1500371,Werre
256+
1416446,Fier
257+
1422413,Francolí
258+
1565684,Stör
259+
1509367,Dijle
260+
1518333,Adour
261+
1504187,Lippe
262+
1504383,Thur
263+
1504240,Sambre
264+
1504778,Glatt
265+
1503876,Meuse
266+
1505020,Sitter
267+
1504777,Birs
268+
1504847,Vesdre
269+
1504653,Semois
270+
1504255,Reuss
271+
1504091,Rhein

0 commit comments

Comments
 (0)