Core Model

StringCheckerModel.scala

package ml.combust.mleap.core.feature
 
import ml.combust.mleap.core.Model
import ml.combust.mleap.core.types.{ScalarType, StructField, StructType}
 
/**
  * Created by mostafam on 6/30/20.
  */
 
case class StringCheckerModel(caseSensitive: Boolean) extends Model {
 
  def apply(text: String, query: String): Double = {
    if (caseSensitive){
      text.contains(query).compare(false)
    } else {
      text.toLowerCase().contains(query.toLowerCase()).compare(false)
    }
  }
 
  override def inputSchema: StructType = StructType(
    StructField(s"text", ScalarType.String),
    StructField(s"query", ScalarType.String)
  ).get
 
  override def outputSchema: StructType = StructType("output" -> ScalarType.Double.nonNullable).get
 
}