Merge pull request #64 from alan-turing-institute/dev

For a 0.3.3 release

Author: ablaom

Project.toml

@@ -1,7 +1,7 @@
 name = "MLJModelInterface"
 uuid = "e80e1ace-859a-464e-9ed9-23947d8ae3ea"
 authors = ["Thibaut Lienart and Anthony Blaom"]
-version = "0.3.2"
+version = "0.3.3"
 
 [deps]
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

src/MLJModelInterface.jl

@@ -14,7 +14,8 @@ export LightInterface, FullInterface
 # MLJ model hierarchy
 export MLJType, Model, Supervised, Unsupervised,
        Probabilistic, Deterministic, Interval, Static,
-       UnivariateFinite
+       UnivariateFinite,
+       JointProbabilistic
 
 # parameter_inspection:
 export params
@@ -88,6 +89,8 @@ abstract type      Interval <: Supervised end
 
 abstract type Static <: Unsupervised end
 
+abstract type JointProbabilistic <: Probabilistic end
+
 # ------------------------------------------------------------------------
 # includes
 

src/model_api.jl

@@ -44,6 +44,13 @@ probabilistic supervised models may overload `predict_median`
 """
 function predict_median end
 
+"""
+`JointProbabilistic` supervised models MUST overload `predict_joint`.
+
+`Probabilistic` supervised models MAY overload `predict_joint`.
+"""
+function predict_joint end
+
 """
 unsupervised methods must implement the `transform` operation
 """