[机器学习]用Java实现梯度下降

最新推荐文章于 2024-09-08 22:03:07 发布

ojama

最新推荐文章于 2024-09-08 22:03:07 发布

阅读量1.5k

点赞数

分类专栏：机器学习文章标签：机器学习梯度下降

本文链接：https://blog.csdn.net/jidong2622/article/details/79342525

版权

本文详细介绍了如何使用Java实现机器学习中的梯度下降算法，基于Coursera上的多元梯度下降教程，涵盖了整个运算过程。

摘要由CSDN通过智能技术生成

这是在coursea的解释多元的梯度下降运算过程

实际也就是求偏导数

本测试用例为2元

但适用于多元的数据

数据如下

1,2,3

1,2,3

代码如下

package hello;

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++