mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-18 09:08:48 +02:00
ef4c5eb65e
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
27 lines
1.1 KiB
Scala
27 lines
1.1 KiB
Scala
package com.twitter.home_mixer.functional_component.decorator
|
|
|
|
import com.twitter.home_mixer.model.HomeFeatures.ScoreFeature
|
|
import com.twitter.home_mixer.param.HomeGlobalParams.EnableSendScoresToClient
|
|
import com.twitter.product_mixer.component_library.model.candidate.TweetCandidate
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMap
|
|
import com.twitter.product_mixer.core.functional_component.decorator.urt.builder.item.tweet.BaseTimelinesScoreInfoBuilder
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.tweet.TimelinesScoreInfo
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
|
|
|
object HomeTimelinesScoreInfoBuilder
|
|
extends BaseTimelinesScoreInfoBuilder[PipelineQuery, TweetCandidate] {
|
|
|
|
private val UndefinedTweetScore = -1.0
|
|
|
|
override def apply(
|
|
query: PipelineQuery,
|
|
candidate: TweetCandidate,
|
|
candidateFeatures: FeatureMap
|
|
): Option[TimelinesScoreInfo] = {
|
|
if (query.params(EnableSendScoresToClient)) {
|
|
val score = candidateFeatures.getOrElse(ScoreFeature, None).getOrElse(UndefinedTweetScore)
|
|
Some(TimelinesScoreInfo(score))
|
|
} else None
|
|
}
|
|
}
|