metrics

Documentation

Index

• func PearsonCorrelation(x, y []float64) float64
• func SpearmanCorrelation(x, y []float64) float64
• type Metrics
  • func CalculateMetrics(predictions, targets []float64) *Metrics

Functions

func PearsonCorrelation

func PearsonCorrelation(x, y []float64) float64

PearsonCorrelation calculates the Pearson correlation coefficient between two slices

func SpearmanCorrelation

func SpearmanCorrelation(x, y []float64) float64

SpearmanCorrelation calculates the Spearman rank correlation coefficient

Types

type Metrics

type Metrics struct {
	PearsonCorrelation  float64
	SpearmanCorrelation float64
	MSE                 float64
	RMSE                float64
	MAE                 float64
}

Metrics holds evaluation metrics for model performance

func CalculateMetrics

func CalculateMetrics(predictions, targets []float64) *Metrics

CalculateMetrics computes evaluation metrics for predictions vs targets