go-train-delay-ml-pipeline-feature-target

go-train-delay-ml-pipeline-feature-target.rs

async fn feature_and_target(
        &self,
        df: &DataFrame,
) -> (PolarResult<DataFrame>, PolarResult<DataFrame>) {
        /* Read a dataframe, select the columns we need for feature training and target and return
        the two new dataframes*/
        //id,date,departure,depart_scheduled,destination,arrival_scheduled

        let features = df.select(vec![
            "date",
            "departure",
            "depart_scheduled",
            "destination",
            "arrival_scheduled",
        ]);

        let target = df.select("delay");

        (features, target)
}