/
githubmirror
/
the-algorithm
Обзор
Документация
Войти
/
githubmirror
/
the-algorithm
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
src/scala/com/twitter/simclusters_v2/common/SimClustersMultiEmbedding.scala
32 строки
1017 B
twitter-team
Twitter Recommendation Algorithm
01 апр 2023, 01:36
01 апр 2023, 01:36
ef4c5eb
Код
Авторство
О чём код?
package com.twitter.simclusters_v2.common import com.twitter.simclusters_v2.common.SimClustersMultiEmbeddingId._ import com.twitter.simclusters_v2.thriftscala.SimClustersMultiEmbedding.{Ids, Values} import com.twitter.simclusters_v2.thriftscala.{ SimClustersMultiEmbedding, SimClustersEmbeddingId, SimClustersMultiEmbeddingId } /** * Helper methods for SimClustersMultiEmbedding */ object SimClustersMultiEmbedding { // Convert a multiEmbedding to a list of (embeddingId, score) def toSimClustersEmbeddingIdWithScores( simClustersMultiEmbeddingId: SimClustersMultiEmbeddingId, simClustersMultiEmbedding: SimClustersMultiEmbedding ): Seq[(SimClustersEmbeddingId, Double)] = { simClustersMultiEmbedding match { case Values(values) => values.embeddings.zipWithIndex.map { case (embeddingWithScore, i) => (toEmbeddingId(simClustersMultiEmbeddingId, i), embeddingWithScore.score) } case Ids(ids) => ids.ids.map(_.toTuple) } } }