• Home
• net.ruippeixotog
• think-bayes_2.11
• 0.1
• source code
• package.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.

thinkbayes.package.scala Maven / Gradle / Ivy

package object thinkbayes {

  implicit class HistogramUtils[K](val hist: Map[K, Double]) extends AnyVal {

    private def pad(str: String, n: Int): String =
      if (str.length > n) str.substring(0, n) else str + (" " * (n - str.length))

    def toPmf = new Pmf(hist)

    def print()(implicit ord: Ordering[K]) {
      if (hist.nonEmpty) {
        val keyLen = hist.keys.map(_.toString.length).max
        hist.toSeq.sortBy(_._1).map {
          case (h, prob) => pad(h.toString, keyLen) + " " + prob
        }.foreach(println)
      }
    }

    def printChart()(implicit ord: Ordering[K]) {
      if (hist.nonEmpty) {
        val keyLen = hist.keys.map(_.toString.length).max
        hist.toSeq.sortBy(_._1).map {
          case (h, prob) =>
            pad(h.toString, keyLen).mkString + " " +
              pad(prob.toString, 6) + " " +
              ("#" * (50 * prob).toInt)
        }.foreach(println)
      }
    }
  }

  implicit def cdfHistogramUtils[K](cdf: Cdf[K]) = new HistogramUtils(cdf.vals.toMap)
  implicit def suiteHistogramUtils[H](suite: Suite[H, _]) = new HistogramUtils(suite.pmf)
}