regression.pro

来自「利用IDl 编写的一元线性回归程序」· PRO 代码 · 共 39 行

;回归模型系数计算:获取影像的反射率,返回回归模型系数a,b
PRO Regression

    x=[0,2,1,5,4,2,6,2,5,7]
    y=[1,1,3,3,4,5,6,7,8,9]
    numb=n_elements(x)

	YY=dblarr(1,numb);--- Array[10]  全部转为列向量
	XX=dblarr(2,numb)
	for i=0,numb-1 DO begin
		YY[0,i]=y[i]
		XX[0,i]=x[i]
		XX[1,i]=1
	endfor



	XXt=transpose(XX);----Array[10, 2]

	C=XXt##XX;----------- Array[2, 2]

	C_1=Invert(C);----- Array[2, 2]


	D=c_1##XXt;---------- Array[10, 2]


	result=D##YY;----Array[1, 2]

	help,result
	num1=result[0,0]
	num2=result[0,1]
	print,"模型系数"
	print,num1,num2
end