24 lines
870 B
Scala
24 lines
870 B
Scala
package com.twitter.frigate.pushservice.config.mlconfig
|
|
|
|
import com.twitter.cortex.deepbird.thriftjava.DeepbirdPredictionService
|
|
import com.twitter.finagle.stats.StatsReceiver
|
|
import com.twitter.frigate.common.ml.prediction.DeepbirdPredictionEngineServiceStore
|
|
import com.twitter.nrel.heavyranker.PushDBv2PredictionServiceStore
|
|
|
|
object DeepbirdV2ModelConfig {
|
|
def buildPredictionServiceScoreStore(
|
|
predictionServiceClient: DeepbirdPredictionService.ServiceToClient,
|
|
serviceName: String
|
|
)(
|
|
implicit statsReceiver: StatsReceiver
|
|
): PushDBv2PredictionServiceStore = {
|
|
|
|
val stats = statsReceiver.scope(serviceName)
|
|
val serviceStats = statsReceiver.scope("dbv2PredictionServiceStore")
|
|
|
|
new PushDBv2PredictionServiceStore(
|
|
DeepbirdPredictionEngineServiceStore(predictionServiceClient, batchSize = Some(32))(stats)
|
|
)(serviceStats)
|
|
}
|
|
}
|