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
ConstructorsConstructorDescriptionSmileClassificationModel(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 TypeMethodDescriptioncopy(org.apache.spark.ml.param.ParamMap extra) static SmileClassificationModelLoads a persisted SmileClassificationModel from the given path.Classifier<double[]> model()Returns the underlying SMILE classifier.intorg.apache.spark.ml.linalg.VectorpredictRaw(org.apache.spark.ml.linalg.Vector features) static org.apache.spark.ml.util.MLReader<SmileClassificationModel> read()Returns an MLReader instance to load a persisted SmileClassificationModel.uid()org.apache.spark.ml.util.MLWriterwrite()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, validateAndTransformSchemaMethods 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, setPredictionColMethods inherited from class org.apache.spark.ml.Model
estimatedSize, hasParent, parent, parent_$eq, setParentMethods inherited from class org.apache.spark.ml.Transformer
transform, transform, transformMethods 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, withLogContextMethods inherited from class Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, waitMethods inherited from interface org.apache.spark.ml.param.shared.HasFeaturesCol
featuresCol, getFeaturesCol, org$apache$spark$ml$param$shared$HasFeaturesCol$_setter_$featuresCol_$eqMethods inherited from interface org.apache.spark.ml.param.shared.HasLabelCol
getLabelCol, labelCol, org$apache$spark$ml$param$shared$HasLabelCol$_setter_$labelCol_$eqMethods inherited from interface org.apache.spark.ml.param.shared.HasPredictionCol
getPredictionCol, org$apache$spark$ml$param$shared$HasPredictionCol$_setter_$predictionCol_$eq, predictionColMethods inherited from interface org.apache.spark.ml.util.Identifiable
toStringMethods inherited from interface org.apache.spark.ml.util.MLWritable
saveMethods 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
Constructor with a default random UID.- Parameters:
numClasses- the number of target classes.model- the underlying SMILE classifier.
-
SmileClassificationModel
Constructor with explicit UID.- Parameters:
uid- the instance UID.numClasses- the number of target classes.model- the underlying SMILE classifier.
-
-
Method Details
-
uid
- Specified by:
uidin interfaceorg.apache.spark.ml.util.Identifiable
-
numClasses
public int numClasses()- Specified by:
numClassesin classorg.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector, SmileClassificationModel>
-
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:
predictRawin classorg.apache.spark.ml.classification.ClassificationModel<org.apache.spark.ml.linalg.Vector, SmileClassificationModel>
-
copy
- Specified by:
copyin interfaceorg.apache.spark.ml.param.Params- Specified by:
copyin classorg.apache.spark.ml.Model<SmileClassificationModel>
-
write
public org.apache.spark.ml.util.MLWriter write()- Specified by:
writein interfaceorg.apache.spark.ml.util.MLWritable
-
read
Returns an MLReader instance to load a persisted SmileClassificationModel.- Returns:
- the reader.
-
load
Loads a persisted SmileClassificationModel from the given path.- Parameters:
path- the path.- Returns:
- the loaded model.
-