Существует ли библиотека Javascript для выполнения Spearman и /или корреляции Пирсона?
Библиотека Javascript для корреляций Пирсона и/или Спирмена
Ответы (5)
Есть это
http://stevegardner.net/2012/06/11/javascript-code-to-calculate-the-pearson-correlation-coefficient/
кроме этого вы можете попробовать:
в качестве альтернативы, если ни один из них не подходит по счету, и вы не хотите писать его самостоятельно, вы всегда можете использовать:
с
http://www.gardenersown.co.uk/education/lectures/r/correl.htm
to do it.
Итак, вот мои две копейки по этому вопросу - корреляция Пирсона:
const pcorr = (x, y) => {
let sumX = 0,
sumY = 0,
sumXY = 0,
sumX2 = 0,
sumY2 = 0;
const minLength = x.length = y.length = Math.min(x.length, y.length),
reduce = (xi, idx) => {
const yi = y[idx];
sumX += xi;
sumY += yi;
sumXY += xi * yi;
sumX2 += xi * xi;
sumY2 += yi * yi;
}
x.forEach(reduce);
return (minLength * sumXY - sumX * sumY) / Math.sqrt((minLength * sumX2 - sumX * sumX) * (minLength * sumY2 - sumY * sumY));
};
let arrX = [20, 54, 54, 65, 45];
let arrY = [22, 11, 21, 34, 87];
let R = pcorr(arrX, arrY);
console.log('arrX', arrX, 'arrY', arrY, 'R', R);
Я использовал проект Spearson здесь на Github. Я проверил его на корреляцию Спирмена, и он дает точные значения для этого.
Я только что скачал файл spearson.js
в папку /lib
репозитория. Вот как использовать его в браузере:
<script src="spearson.js"></script>
<script>
var x = [3, 4, 5];
var y = [.1, .2, .3];
var corr = spearson.correlation.spearman(x, y);
</script>
Точно так же вы можете использовать correlation.pearson
для корреляции Пирсона.
Попробуй это:
function spearmanCorrelation(multiList, p1, p2){
N=multiList[p1].length;
order=[];
sum=0;
for(i=0;i<N;i++){
order.push([multiList[p1][i], multiList[p2][i]]);
}
order.sort(function(a,b){
return a[0]-b[0]
});
for(i=0;i<N;i++){
order[i].push(i+1);
}
order.sort(function(a,b){
return a[1]-b[1]
});
for(i=0;i<N;i++){
order[i].push(i+1);
}
for(i=0;i<N;i++){
sum+=Math.pow((order[i][2])-(order[i][3]), 2);
}
r=1-(6*sum/(N*(N*N-1)));
return r;
}
Спустя много лет после того, как был задан первоначальный вопрос, я рекомендую эту замечательную библиотеку, которая проста, но все еще хорошо документирована: statistics.js
Среди многих других методов у него есть correlationCoefficient(), который вычисляет Коэффициент корреляции Пирсона двух переменных и spearmansRho() который определяет коэффициент ранговой корреляции Спирмена.