• Home
• com.iheart
• thomas-analysis_2.13
• 1.4.8
• source code
• BetaModel.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.iheart.thomas.analysis.bayesian.models.BetaModel.scala Maven / Gradle / Ivy

package com.iheart.thomas.analysis
package bayesian.models
import syntax.all._
import cats.data.ValidatedNel
import com.stripe.rainier.core.Beta
import cats.implicits._
case class BetaModel(
    alpha: Double,
    beta: Double) {
  lazy val prediction = Beta(alpha, beta).latent

  override def toString: String = {
    val converted = alpha - 1d
    val total = beta - 1 + converted
    s"Beta Model based on converted: $converted total: $total"
  }
}

object BetaModel {
  def apply(conversions: Conversions): BetaModel =
    BetaModel(
      alpha = conversions.converted + 1d,
      beta = conversions.total - conversions.converted + 1d
    )

  def validate(model: BetaModel): ValidatedNel[String, BetaModel] =
    (model.beta > 0).toValidatedNel(model, "Beta must be larger than zero") <*
      (model.alpha > 0).toValidatedNel(model, "Alpha must be larger than zero")

}