Method write_csv

perfect/src/stats.rs:337–383

Write this set of results to a CSV file. FIXME: This code is very slow and bad. Would probably be better if this structure was reorganized. FIXME: For now, this simply takes the *mode* from each set of observations.

(&self, filename: &str)

    /// observations. 
    ///
    pub fn write_csv(&self, filename: &str) { 
        let mut w = csv::Writer::from_path(filename).unwrap();

        // The first row is a set of labels for each column
        let mut fields = Vec::new();
        fields.push("input".to_string());
        for evt in self.data.keys() { 
            fields.push(evt.as_desc().name().to_string());
        }


        // FIXME: This doesn't validate that all results contain the 
        // same number of observations 
        let first = self.data.iter().next().unwrap();
        let mut num_samples = first.1.len();
        let num_columns = self.data.keys().len();

        let mut rows: Vec<(I, Vec<usize>)> = {
            vec![(I::default(), Vec::new()); num_samples]
        };

        // Fill out the dependent variable column. 
        // FIXME: We only look at one event, but there's no guarantee that 
        // these are the same across all recorded events. 
        for idx in 0..num_samples { 
            rows[idx].0 = first.1.inputs[idx];
        }

        // Fill out the column for each event
        for (evt, results) in self.data.iter() { 
            for idx in 0..num_samples { 
                rows[idx].1 = self.iter_ith_results(idx)
                    .map(|x| x.get_mode())
                    .collect();
            }
        }

        // Write all of the rows
        w.write_record(fields).unwrap();
        for row in rows { 
            let mut r = Vec::new();
            r.push(format!("{}", row.0));
            r.extend(row.1.iter().map(|x| format!("{}", x)));
            w.write_record(r).unwrap();
        }

    }
}

/// Set of results associated with a particular experiment.