feat: adds F1 and roc_auc_score

This commit is contained in:
Volodymyr Orlov
2020-06-08 14:47:59 -07:00
parent ee581abf70
commit 14113b4152
10 changed files with 167 additions and 6 deletions
+10
View File
@@ -27,6 +27,10 @@ impl<T: FloatExt> BaseVector<T> for ArrayBase<OwnedRepr<T>, Ix1> {
fn len(&self) -> usize {
self.len()
}
fn to_vec(&self) -> Vec<T> {
self.to_owned().to_vec()
}
}
impl<T: FloatExt + ScalarOperand + AddAssign + SubAssign + MulAssign + DivAssign + Sum>
@@ -351,6 +355,12 @@ mod tests {
assert_eq!(3, v.len());
}
#[test]
fn vec_to_vec() {
let v = arr1(&[1., 2., 3.]);
assert_eq!(vec![1., 2., 3.], v.to_vec());
}
#[test]
fn from_to_row_vec() {
let vec = arr1(&[1., 2., 3.]);