Merge pull request #167 from pat-alt/probabilisitic-set

Probabilisitic set

Author: ablaom

src/MLJModelInterface.jl

@@ -41,6 +41,7 @@ const ABSTRACT_MODEL_SUBTYPES = [
     :Probabilistic,
     :Deterministic,
     :Interval,
+    :ProbabilisticSet,
     :JointProbabilistic,
     :Static,
     :Annotator,
@@ -143,6 +144,7 @@ abstract type Annotator <: Model end
 abstract type Probabilistic <: Supervised end
 abstract type Deterministic <: Supervised end
 abstract type Interval <: Supervised end
+abstract type ProbabilisticSet <: Supervised end
 
 abstract type JointProbabilistic <: Probabilistic end
 

src/model_traits.jl

@@ -27,6 +27,7 @@ StatTraits.is_supervised(::Type{<:SupervisedAnnotator}) = true
 StatTraits.prediction_type(::Type{<:Deterministic}) = :deterministic
 StatTraits.prediction_type(::Type{<:Probabilistic}) = :probabilistic
 StatTraits.prediction_type(::Type{<:Interval}) = :interval
+StatTraits.prediction_type(::Type{<:ProbabilisticSet}) = :probabilistic_set
 StatTraits.prediction_type(::Type{<:ProbabilisticDetector}) = :probabilistic
 StatTraits.prediction_type(::Type{<:DeterministicDetector}) = :deterministic