Function create_classification_model 

pub fn create_classification_model(
    _input_dim: usize,
    num_classes: usize,
    hidden_layers: Vec<usize>,
) -> Sequential

Create a simple sequential model for classification