predict.ts

export const getFavouriteCluster = async (songIds: ObjectId[]) => {
    const db = getDb();

    // Since there are 8 total clusters for the model
    const clusters: number[] = new Array(8).fill(0);
    const collection = db.collection<Song>("song");

    // Get the cluster from each favourite song
    const songs = await collection
        .find({ _id: { $in: songIds } }, { projection: { cluster: 1 } })
        .toArray();

    // Keep track of the number of songs with a certain cluster
    for (const song of songs) {
        clusters[song.cluster]++;
    }

    // Remember that each array index corresponds to the cluster number
    const favouriteCluster = clusters.indexOf(Math.max(...clusters));
    return favouriteCluster;
};