mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-02 01:08:47 +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.
67 lines
2.8 KiB
Scala
67 lines
2.8 KiB
Scala
package com.twitter.product_mixer.component_library.feature.featurestorev1
|
|
|
|
import com.twitter.ml.api.transform.FeatureRenameTransform
|
|
import com.twitter.ml.featurestore.catalog.entities
|
|
import com.twitter.ml.featurestore.lib.EdgeEntityId
|
|
import com.twitter.ml.featurestore.lib.EntityId
|
|
import com.twitter.ml.featurestore.lib.TweetId
|
|
import com.twitter.ml.featurestore.lib.UserId
|
|
import com.twitter.ml.featurestore.lib.entity.Entity
|
|
import com.twitter.ml.featurestore.lib.entity.EntityWithId
|
|
import com.twitter.ml.featurestore.lib.feature.TimelinesAggregationFrameworkFeatureGroup
|
|
import com.twitter.ml.featurestore.lib.feature.{Feature => FSv1Feature}
|
|
import com.twitter.product_mixer.component_library.model.candidate.BaseTweetCandidate
|
|
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.feature.featurestorev1._
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
|
import com.twitter.timelines.configapi.FSParam
|
|
import scala.reflect.ClassTag
|
|
|
|
object FeatureStoreV1QueryUserIdTweetCandidateTweetIdFeature {
|
|
def apply[Query <: PipelineQuery, Candidate <: BaseTweetCandidate, Value](
|
|
feature: FSv1Feature[EdgeEntityId[UserId, TweetId], Value],
|
|
legacyName: Option[String] = None,
|
|
defaultValue: Option[Value] = None,
|
|
enabledParam: Option[FSParam[Boolean]] = None
|
|
): FeatureStoreV1CandidateFeature[Query, Candidate, _ <: EntityId, Value] =
|
|
FeatureStoreV1CandidateFeature(
|
|
feature,
|
|
QueryUserIdTweetCandidateTweetIdEntity,
|
|
legacyName,
|
|
defaultValue,
|
|
enabledParam)
|
|
}
|
|
|
|
object FeatureStoreV1QueryUserIdTweetCandidateTweetIdAggregateFeature {
|
|
def apply[Query <: PipelineQuery, Candidate <: BaseTweetCandidate](
|
|
featureGroup: TimelinesAggregationFrameworkFeatureGroup[EdgeEntityId[UserId, TweetId]],
|
|
enabledParam: Option[FSParam[Boolean]] = None,
|
|
keepLegacyNames: Boolean = false,
|
|
featureNameTransform: Option[FeatureRenameTransform] = None
|
|
): FeatureStoreV1CandidateFeatureGroup[Query, TweetCandidate, _ <: EntityId] =
|
|
FeatureStoreV1CandidateFeatureGroup(
|
|
featureGroup,
|
|
QueryUserIdTweetCandidateTweetIdEntity,
|
|
enabledParam,
|
|
keepLegacyNames,
|
|
featureNameTransform
|
|
)(implicitly[ClassTag[EdgeEntityId[UserId, TweetId]]])
|
|
}
|
|
|
|
object QueryUserIdTweetCandidateTweetIdEntity
|
|
extends FeatureStoreV1CandidateEntity[
|
|
PipelineQuery,
|
|
BaseTweetCandidate,
|
|
EdgeEntityId[UserId, TweetId]
|
|
] {
|
|
override val entity: Entity[EdgeEntityId[UserId, TweetId]] = entities.core.UserTweet
|
|
|
|
override def entityWithId(
|
|
query: PipelineQuery,
|
|
tweet: BaseTweetCandidate,
|
|
existingFeatures: FeatureMap
|
|
): EntityWithId[EdgeEntityId[UserId, TweetId]] =
|
|
entity.withId(EdgeEntityId(UserId(query.getUserIdLoggedOutSupport), TweetId(tweet.id)))
|
|
}
|