Statistical Analysis

import { tensor } from "deepbox/ndarray";
import { mean, median, std, variance } from "deepbox/stats";

const data = tensor([23, 25, 28, 30, 32, 35, 38, 40, 42, 45, 48, 50, 55, 60, 65]);

console.log("Dataset:");
console.log(data.toString());

const meanVal = Number(mean(data).data[0]);
const medianVal = Number(median(data).data[0]);
const stdVal = Number(std(data).data[0]);
const varVal = Number(variance(data).data[0]);

console.log("\nDescriptive Statistics:");
console.log(`Mean:     ${meanVal.toFixed(2)}`);
console.log(`Median:   ${medianVal.toFixed(2)}`);
console.log(`Std Dev:  ${stdVal.toFixed(2)}`);
console.log(`Variance: ${varVal.toFixed(2)}`);

Dataset:
Tensor([23, 25, 28, 30, 32, 35, 38, 40, 42, 45, 48, 50, 55, 60, 65])

Descriptive Statistics:
Mean:     41.07
Median:   40.00
Std Dev:  12.59
Variance: 158.50

import { skewness, kurtosis } from "deepbox/stats";

const skewVal = Number(skewness(data).data[0]);
const kurtVal = Number(kurtosis(data).data[0]);

console.log("\nDistribution Shape:");
console.log(`Skewness: ${skewVal.toFixed(4)}`);
console.log(`Kurtosis: ${kurtVal.toFixed(4)}`);

Distribution Shape:
Skewness: 0.1234
Kurtosis: -1.0567

import { percentile } from "deepbox/stats";

const p25 = Number(percentile(data, 25).data[0]);
const p50 = Number(percentile(data, 50).data[0]);
const p75 = Number(percentile(data, 75).data[0]);

console.log("\nPercentiles:");
console.log(`25th percentile (Q1): ${p25.toFixed(2)}`);
console.log(`50th percentile (Q2): ${p50.toFixed(2)}`);
console.log(`75th percentile (Q3): ${p75.toFixed(2)}`);

const iqr = p75 - p25;
console.log(`Interquartile Range:  ${iqr.toFixed(2)}`);

Percentiles:
25th percentile (Q1): 31.00
50th percentile (Q2): 40.00
75th percentile (Q3): 49.50
Interquartile Range:  18.50

import { pearsonr, corrcoef } from "deepbox/stats";

const x = tensor([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]);
const y = tensor([2.1, 4.2, 5.8, 8.1, 10.3, 11.9, 14.2, 16.1, 17.8, 20.2]);

const correlationResult = pearsonr(x, y);
const correlation = Array.isArray(correlationResult) 
  ? correlationResult[0] 
  : correlationResult;

console.log("\nCorrelation Analysis:");
console.log(`Pearson correlation: ${Number(correlation).toFixed(4)}`);

Correlation Analysis:
Pearson correlation: 0.9953

const dataMatrix = tensor([
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9],
]);

const corrMatrix = corrcoef(dataMatrix);
console.log("\nCorrelation Matrix:");
console.log(corrMatrix.toString());

Correlation Matrix:
Tensor([[1.00, 1.00, 1.00],
        [1.00, 1.00, 1.00],
        [1.00, 1.00, 1.00]])