Class SmileClassificationModel

java.lang.Object
org.apache.spark.ml.PipelineStage
org.apache.spark.ml.Transformer
org.apache.spark.ml.Model<SmileClassificationModel>
org.apache.spark.ml.PredictionModel<org.apache.spark.ml.linalg.Vector, SmileClassificationModel>
org.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector, SmileClassificationModel>
smile.spark.SmileClassificationModel
All Implemented Interfaces:
Serializable, org.apache.spark.internal.Logging, org.apache.spark.ml.classification.ClassifierParams, org.apache.spark.ml.param.Params, org.apache.spark.ml.param.shared.HasFeaturesCol, org.apache.spark.ml.param.shared.HasLabelCol, org.apache.spark.ml.param.shared.HasPredictionCol, org.apache.spark.ml.param.shared.HasRawPredictionCol, org.apache.spark.ml.PredictorParams, org.apache.spark.ml.util.Identifiable, org.apache.spark.ml.util.MLWritable

public class SmileClassificationModel extends org.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector, SmileClassificationModel> implements org.apache.spark.ml.util.MLWritable
A Spark ML Model produced by SmileClassifier.
See Also:
  • Nested Class Summary

    Nested classes/interfaces inherited from interface org.apache.spark.internal.Logging

    org.apache.spark.internal.Logging.LogStringContext, org.apache.spark.internal.Logging.SparkShellLoggingFilter
  • Constructor Summary

    Constructors
    Constructor
    Description
    SmileClassificationModel(int numClasses, Classifier<double[]> model)
    Constructor with a default random UID.
    SmileClassificationModel(String uid, int numClasses, Classifier<double[]> model)
    Constructor with explicit UID.
  • Method Summary

    Modifier and Type
    Method
    Description
    copy(org.apache.spark.ml.param.ParamMap extra)
     
    load(String path)
    Loads a persisted SmileClassificationModel from the given path.
    Classifier<double[]>
    Returns the underlying SMILE classifier.
    int
     
    org.apache.spark.ml.linalg.Vector
    predictRaw(org.apache.spark.ml.linalg.Vector features)
     
    static org.apache.spark.ml.util.MLReader<SmileClassificationModel>
    Returns an MLReader instance to load a persisted SmileClassificationModel.
    uid()
     
    org.apache.spark.ml.util.MLWriter
     

    Methods inherited from class org.apache.spark.ml.classification.ClassificationModel

    findSummaryModel, getRawPredictionCol, org$apache$spark$ml$param$shared$HasRawPredictionCol$_setter_$rawPredictionCol_$eq, predict, raw2prediction, rawPredictionCol, setRawPredictionCol, transform, transformImpl, transformSchema, validateAndTransformSchema

    Methods inherited from class org.apache.spark.ml.PredictionModel

    featuresCol, featuresDataType, getFeaturesCol, getLabelCol, getPredictionCol, labelCol, numFeatures, org$apache$spark$ml$param$shared$HasFeaturesCol$_setter_$featuresCol_$eq, org$apache$spark$ml$param$shared$HasLabelCol$_setter_$labelCol_$eq, org$apache$spark$ml$param$shared$HasPredictionCol$_setter_$predictionCol_$eq, predictionCol, setFeaturesCol, setPredictionCol

    Methods inherited from class org.apache.spark.ml.Model

    estimatedSize, hasParent, parent, parent_$eq, setParent

    Methods inherited from class org.apache.spark.ml.Transformer

    transform, transform, transform

    Methods inherited from class org.apache.spark.ml.PipelineStage

    $, clear, copyValues, copyValues$default$2, defaultCopy, defaultParamMap, estimateMatadataSize, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, initializeForcefully, initializeLogIfNecessary, initializeLogIfNecessary, initializeLogIfNecessary$default$2, isDefined, isSet, isTraceEnabled, log, logBasedOnLevel, logDebug, logDebug, logDebug, logDebug, logError, logError, logError, logError, logInfo, logInfo, logInfo, logInfo, logName, LogStringContext, logTrace, logTrace, logTrace, logTrace, logWarning, logWarning, logWarning, logWarning, MDC, onParamChange, org$apache$spark$internal$Logging$$log_, org$apache$spark$internal$Logging$$log__$eq, org$apache$spark$ml$param$Params$_setter_$defaultParamMap_$eq, org$apache$spark$ml$param$Params$_setter_$paramMap_$eq, paramMap, params, set, set, set, setDefault, setDefault, toString, transformSchema, withLogContext

    Methods inherited from class Object

    clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

    Methods inherited from interface org.apache.spark.ml.param.shared.HasFeaturesCol

    featuresCol, getFeaturesCol, org$apache$spark$ml$param$shared$HasFeaturesCol$_setter_$featuresCol_$eq

    Methods inherited from interface org.apache.spark.ml.param.shared.HasLabelCol

    getLabelCol, labelCol, org$apache$spark$ml$param$shared$HasLabelCol$_setter_$labelCol_$eq

    Methods inherited from interface org.apache.spark.ml.param.shared.HasPredictionCol

    getPredictionCol, org$apache$spark$ml$param$shared$HasPredictionCol$_setter_$predictionCol_$eq, predictionCol

    Methods inherited from interface org.apache.spark.ml.util.Identifiable

    toString

    Methods inherited from interface org.apache.spark.ml.util.MLWritable

    save

    Methods inherited from interface org.apache.spark.ml.param.Params

    $, clear, copyValues, copyValues$default$2, defaultCopy, defaultParamMap, estimateMatadataSize, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, onParamChange, org$apache$spark$ml$param$Params$_setter_$defaultParamMap_$eq, org$apache$spark$ml$param$Params$_setter_$paramMap_$eq, paramMap, params, set, set, set, setDefault, setDefault
  • Constructor Details

    • SmileClassificationModel

      public SmileClassificationModel(int numClasses, Classifier<double[]> model)
      Constructor with a default random UID.
      Parameters:
      numClasses - the number of target classes.
      model - the underlying SMILE classifier.
    • SmileClassificationModel

      public SmileClassificationModel(String uid, int numClasses, Classifier<double[]> model)
      Constructor with explicit UID.
      Parameters:
      uid - the instance UID.
      numClasses - the number of target classes.
      model - the underlying SMILE classifier.
  • Method Details

    • uid

      public String uid()
      Specified by:
      uid in interface org.apache.spark.ml.util.Identifiable
    • numClasses

      public int numClasses()
      Specified by:
      numClasses in class org.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector, SmileClassificationModel>
    • model

      public Classifier<double[]> model()
      Returns the underlying SMILE classifier.
      Returns:
      the SMILE classifier.
    • predictRaw

      public org.apache.spark.ml.linalg.Vector predictRaw(org.apache.spark.ml.linalg.Vector features)
      Specified by:
      predictRaw in class org.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector, SmileClassificationModel>
    • copy

      public SmileClassificationModel copy(org.apache.spark.ml.param.ParamMap extra)
      Specified by:
      copy in interface org.apache.spark.ml.param.Params
      Specified by:
      copy in class org.apache.spark.ml.Model<SmileClassificationModel>
    • write

      public org.apache.spark.ml.util.MLWriter write()
      Specified by:
      write in interface org.apache.spark.ml.util.MLWritable
    • read

      public static org.apache.spark.ml.util.MLReader<SmileClassificationModel> read()
      Returns an MLReader instance to load a persisted SmileClassificationModel.
      Returns:
      the reader.
    • load

      public static SmileClassificationModel load(String path)
      Loads a persisted SmileClassificationModel from the given path.
      Parameters:
      path - the path.
      Returns:
      the loaded model.