use ndarray::prelude::*;
use tangram_table::prelude::*;
use tangram_zip::zip;
#[derive(Clone, Debug)]
pub struct OneHotEncodedFeatureGroup {
pub source_column_name: String,
pub variants: Vec<String>,
}
impl OneHotEncodedFeatureGroup {
pub fn compute_for_column(column: TableColumnView) -> OneHotEncodedFeatureGroup {
match column {
TableColumnView::Enum(column) => Self::compute_for_enum_column(column),
_ => unimplemented!(),
}
}
fn compute_for_enum_column(column: EnumTableColumnView) -> Self {
Self {
source_column_name: column.name().unwrap().to_owned(),
variants: column.variants().to_owned(),
}
}
}
impl OneHotEncodedFeatureGroup {
pub fn compute_array_f32(
&self,
features: ArrayViewMut2<f32>,
column: TableColumnView,
progress: &impl Fn(),
) {
match column {
TableColumnView::Enum(column) => {
self.compute_array_f32_for_enum_column(features, column, progress)
}
TableColumnView::Unknown(_) => unimplemented!(),
TableColumnView::Number(_) => unimplemented!(),
TableColumnView::Text(_) => unimplemented!(),
}
}
fn compute_array_f32_for_enum_column(
&self,
mut features: ArrayViewMut2<f32>,
column: EnumTableColumnView,
progress: &impl Fn(),
) {
features.fill(0.0);
for (mut features, value) in zip!(features.axis_iter_mut(Axis(0)), column.as_slice().iter())
{
let feature_index = value.map(|v| v.get()).unwrap_or(0);
features[feature_index] = 1.0;
progress();
}
}
}