mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-01 08:48:46 +02:00
25 lines
815 B
Scala
25 lines
815 B
Scala
|
package com.twitter.timelines.prediction.features.list_features
|
||
|
|
||
|
import com.twitter.ml.api.Feature.{Binary, Discrete}
|
||
|
import com.twitter.ml.api.FeatureContext
|
||
|
import com.twitter.dal.personal_data.thriftjava.PersonalDataType._
|
||
|
import scala.collection.JavaConverters._
|
||
|
|
||
|
object ListFeatures {
|
||
|
|
||
|
// list.id is used for list tweet injections in home. timelines.meta.list_id is used for list tweets in list timeline.
|
||
|
val LIST_ID = new Discrete("list.id")
|
||
|
|
||
|
val VIEWER_IS_OWNER =
|
||
|
new Binary("list.viewer.is_owner", Set(ListsNonpublicList, ListsPublicList).asJava)
|
||
|
val VIEWER_IS_SUBSCRIBER = new Binary("list.viewer.is_subscriber")
|
||
|
val IS_PINNED_LIST = new Binary("list.is_pinned")
|
||
|
|
||
|
val featureContext = new FeatureContext(
|
||
|
LIST_ID,
|
||
|
VIEWER_IS_OWNER,
|
||
|
VIEWER_IS_SUBSCRIBER,
|
||
|
IS_PINNED_LIST
|
||
|
)
|
||
|
}
|