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
+1 -1
View File
@@ -239,7 +239,7 @@ impl<T: FloatExt> DecisionTreeClassifier<T> {
let mut order: Vec<Vec<usize>> = Vec::new();
for i in 0..num_attributes {
order.push(x.get_col_as_vec(i).quick_argsort());
order.push(x.get_col_as_vec(i).quick_argsort_mut());
}
let mut tree = DecisionTreeClassifier {
+1 -1
View File
@@ -164,7 +164,7 @@ impl<T: FloatExt> DecisionTreeRegressor<T> {
let mut order: Vec<Vec<usize>> = Vec::new();
for i in 0..num_attributes {
order.push(x.get_col_as_vec(i).quick_argsort());
order.push(x.get_col_as_vec(i).quick_argsort_mut());
}
let mut tree = DecisionTreeRegressor {