feat: adds train/test split function; fixes bug in random forest

This commit is contained in:
Volodymyr Orlov
2020-09-13 16:23:30 -07:00
parent 1920f9cd0b
commit d28f13d849
9 changed files with 187 additions and 10 deletions
+12
View File
@@ -72,6 +72,18 @@ impl<T: RealNumber> BaseVector<T> for ArrayBase<OwnedRepr<T>, Ix1> {
fn to_vec(&self) -> Vec<T> {
self.to_owned().to_vec()
}
fn zeros(len: usize) -> Self {
Array::zeros(len)
}
fn ones(len: usize) -> Self {
Array::ones(len)
}
fn fill(len: usize, value: T) -> Self {
Array::from_elem(len, value)
}
}
impl<T: RealNumber + ScalarOperand + AddAssign + SubAssign + MulAssign + DivAssign + Sum>