utils

type ApproxParam struct {
	A   float64 `json:"a"`
	B   float64 `json:"b"`
	Deg int
}

type ApproxParams struct {
	Params []ApproxParam `json:"intervals"`
}

type Bias struct {
	B   []float64 `json:"b"`
	Len int       `json:"len"`
}

type ChebyPolyApprox struct {
	PolyApprox
	A, B      float64
	Degree    int
	Poly      *ckks.Polynomial
	ChebyBase bool
	F         func(x float64) float64
}

type Kernel struct {
	W    []float64 `json:"w"`
	Rows int       `json:"rows"`
	Cols int       `json:"cols"`
}

type Layer struct {
	Weight Kernel `json:"weight"`
	Bias   Bias   `json:"bias"`
}

type PolyApprox interface {
	LevelsOfAct() int
	Rescale(w, b *mat.Dense) (*mat.Dense, *mat.Dense)
}

type Stats struct {
	Iters    int
	Batch    int
	Corrects []int
	Accuracy []float64
	Time     []int64 // Add slice to store time for each evaluation
}