use linregress::{FormulaRegressionBuilder, RegressionDataBuilder};
fn main() {
let y = vec![5., 7., 9., 11.5, 13.];
let x1 = vec![1., 2., 3., 4., 5.];
let data = vec![("Y", y), ("X1", x1)];
let data = RegressionDataBuilder::new().build_from(data).unwrap();
let formula = "Y ~ X1";
let model = FormulaRegressionBuilder::new()
.data(&data)
.formula(formula)
.fit()
.unwrap();
let parameters: Vec<_> = model.iter_parameter_pairs().collect();
let pvalues: Vec<_> = model.iter_p_value_pairs().collect();
let standard_errors: Vec<_> = model.iter_se_pairs().collect();
println!("{:?}", parameters);
println!("{:?}", pvalues);
println!("{:?}", standard_errors);
let others = model.predict(vec![("X1", vec![0., 6.])]).unwrap();
println!("{:?}", others);
}