Adding Multi-Task ElasticNet support

algorithms/linfa-elasticnet/Cargo.toml

@@ -40,6 +40,6 @@ thiserror = "1.0"
 linfa = { version = "0.6.0", path = "../.." }
 
 [dev-dependencies]
-linfa-datasets = { version = "0.6.0", path = "../../datasets", features = ["diabetes"] }
+linfa-datasets = { version = "0.6.0", path = "../../datasets", features = ["diabetes", "linnerud"] }
 ndarray-rand = "0.14"
 rand_xoshiro = "0.6"

algorithms/linfa-elasticnet/examples/elasticnet.rs

@@ -5,7 +5,7 @@ fn main() -> Result<()> {
     // load Diabetes dataset
     let (train, valid) = linfa_datasets::diabetes().split_with_ratio(0.90);
 
-    // train pure LASSO model with 0.1 penalty
+    // train pure LASSO model with 0.3 penalty
     let model = ElasticNet::params()
         .penalty(0.3)
         .l1_ratio(1.0)

algorithms/linfa-elasticnet/examples/multitask_elasticnet.rs

@@ -0,0 +1,24 @@
+use linfa::prelude::*;
+use linfa_elasticnet::{MultiTaskElasticNet, Result};
+
+fn main() -> Result<()> {
+    // load Diabetes dataset
+    let (train, valid) = linfa_datasets::linnerud().split_with_ratio(0.90);
+
+    // train pure LASSO model with 0.1 penalty
+    let model = MultiTaskElasticNet::params()
+        .penalty(0.1)
+        .l1_ratio(1.0)
+        .fit(&train)?;
+
+    println!("intercept:  {}", model.intercept());
+    println!("params: {}", model.hyperplane());
+
+    println!("z score: {:?}", model.z_score());
+
+    // validate
+    let y_est = model.predict(&valid);
+    println!("predicted variance: {}", valid.r2(&y_est)?);
+
+    Ok(())
+}