Function select_features

pub fn select_features(
    batch: RecordBatch,
    input_cols: Vec<&str>,
    output_col: &str,
) -> (NDArray<f64>, NDArray<f64>)

Select specific features of dataset for training