feat: adds F1 and roc_auc_score
This commit is contained in:
@@ -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 {
|
||||
|
||||
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user