mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-01 08:48:46 +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.
32 lines
1.5 KiB
Scala
32 lines
1.5 KiB
Scala
package com.twitter.home_mixer.product.scored_tweets.response_transformer
|
|
|
|
import com.twitter.home_mixer.model.HomeFeatures.CandidateSourceIdFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.SuggestTypeFeature
|
|
import com.twitter.product_mixer.core.feature.Feature
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMap
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMapBuilder
|
|
import com.twitter.product_mixer.core.functional_component.transformer.CandidateFeatureTransformer
|
|
import com.twitter.product_mixer.core.model.common.identifier.TransformerIdentifier
|
|
import com.twitter.timelineranker.{thriftscala => tlr}
|
|
import com.twitter.timelineservice.suggests.logging.candidate_tweet_source_id.{thriftscala => cts}
|
|
import com.twitter.timelineservice.suggests.{thriftscala => st}
|
|
|
|
object ScoredTweetsUtegResponseFeatureTransformer
|
|
extends CandidateFeatureTransformer[tlr.CandidateTweet] {
|
|
|
|
override val identifier: TransformerIdentifier = TransformerIdentifier("ScoredTweetsUtegResponse")
|
|
|
|
override val features: Set[Feature[_, _]] = TimelineRankerResponseTransformer.features
|
|
|
|
override def transform(candidate: tlr.CandidateTweet): FeatureMap = {
|
|
val baseFeatures = TimelineRankerResponseTransformer.transform(candidate)
|
|
|
|
val features = FeatureMapBuilder()
|
|
.add(CandidateSourceIdFeature, Some(cts.CandidateTweetSourceId.RecommendedTweet))
|
|
.add(SuggestTypeFeature, Some(st.SuggestType.ActivityTweet))
|
|
.build()
|
|
|
|
baseFeatures ++ features
|
|
}
|
|
}
|