[机器学习]关于分类问题的梯度下降

在分类问题中 假设有所变化,1+e的-z次方.

这个z就相当于k*x+b*1

所以对于线性回归的问题,梯度下降得做出改变(改变并不大)

X数据值

1,2,3,4,5,6,7,8,9,10,100

Y数据值

0,0,0,0,1,1,1,1,1,1

代码如下

package ojama;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.List;
import java.util.Vector;

public class GradientDescent {
	public static Double[] getTheta(List<Double[]> X, Double[] y) {
		// 初始化长度
		int m = y.length;
		// 初始化theta
		Double[] theta = new Double[X.size()];
		double a = 0.001;
		for (int i = 0; i < theta.length; i++) {
			theta[i] = 0.0;
		}