Merge pull request #23 from alan-turing-institute/master

merge master into dev

Author: ablaom

Project.toml

@@ -1,7 +1,7 @@
 name = "MLJMultivariateStatsInterface"
 uuid = "1b6a4a23-ba22-4f51-9698-8599985d3728"
 authors = ["Anthony D. Blaom <[email protected]>", "Thibaut Lienart <[email protected]>", "Okon Samuel <[email protected]>"]
-version = "0.1.8"
+version = "0.2.1"
 
 [deps]
 Distances = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7"

src/MLJMultivariateStatsInterface.jl

@@ -123,9 +123,19 @@ include("utils.jl")
 # ===================================================================
 # List of all models interfaced
 const MODELS = (
-    LinearRegressor, RidgeRegressor, PCA, KernelPCA, ICA, LDA,
-    BayesianLDA, SubspaceLDA,
-    BayesianSubspaceLDA, FactorAnalysis, PPCA
+    LinearRegressor,
+    MultitargetLinearRegressor,
+    RidgeRegressor,
+    MultitargetRidgeRegressor,
+    PCA,
+    KernelPCA,
+    ICA,
+    LDA,
+    BayesianLDA,
+    SubspaceLDA,
+    BayesianSubspaceLDA,
+    FactorAnalysis,
+    PPCA
 )
 
 # ====================================================================

src/models/discriminant_analysis.jl

@@ -70,8 +70,8 @@ end
 function _check_lda_data(model, X, y)
     class_list = MMI.classes(y[1]) # Class list containing entries in pool of y.
     nclasses = length(class_list)
-    # Class list containing unique entries in y.
-    classes_seen = filter(in(unique(y)), class_list) 
+    # Class list containing entries in seen in y.
+    classes_seen = filter(in(y), class_list) 
     nc = length(classes_seen) # Number of classes in pool of y.
     integers_seen = MMI.int(classes_seen)
     # NOTE: copy/transpose.