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.
41 lines
1.6 KiB
Scala
41 lines
1.6 KiB
Scala
package com.twitter.product_mixer.component_library.feature.featurestorev1
|
|
|
|
import com.twitter.ml.featurestore.catalog.entities
|
|
import com.twitter.ml.featurestore.lib.EntityId
|
|
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.{Feature => FSv1Feature}
|
|
import com.twitter.product_mixer.component_library.model.candidate.BaseUserCandidate
|
|
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
|
|
|
|
object FeatureStoreV1UserCandidateUserIdFeature {
|
|
def apply[Query <: PipelineQuery, Candidate <: BaseUserCandidate, Value](
|
|
feature: FSv1Feature[UserId, Value],
|
|
legacyName: Option[String] = None,
|
|
defaultValue: Option[Value] = None,
|
|
enabledParam: Option[FSParam[Boolean]] = None
|
|
): FeatureStoreV1CandidateFeature[Query, Candidate, _ <: EntityId, Value] =
|
|
FeatureStoreV1CandidateFeature(
|
|
feature,
|
|
UserCandidateUserIdEntity,
|
|
legacyName,
|
|
defaultValue,
|
|
enabledParam)
|
|
}
|
|
|
|
object UserCandidateUserIdEntity
|
|
extends FeatureStoreV1CandidateEntity[PipelineQuery, BaseUserCandidate, UserId] {
|
|
override val entity: Entity[UserId] = entities.core.User
|
|
|
|
override def entityWithId(
|
|
query: PipelineQuery,
|
|
user: BaseUserCandidate,
|
|
existingFeatures: FeatureMap
|
|
): EntityWithId[UserId] =
|
|
entity.withId(UserId(user.id))
|
|
}
|