| 注册
首页|期刊导航|苏州科技大学学报(自然科学版)|梯度下降法在机器学习中的应用

梯度下降法在机器学习中的应用

孙娅楠 林文斌

苏州科技大学学报(自然科学版)2018,Vol.35Issue(2):26-31,6.
苏州科技大学学报(自然科学版)2018,Vol.35Issue(2):26-31,6.DOI:10.12084/j.issn.2096-3289.2018.02.006

梯度下降法在机器学习中的应用

Application of gradient descent method in machine learning

孙娅楠 1林文斌1

作者信息

  • 1. 西南交通大学 数学学院,四川 成都 611756
  • 折叠

摘要

Abstract

In order to solve the problem of machine learning training algorithm ,the author discussed the basic iterative steps of gradient descent method and its variant algorithm. The gradient descent method was used to minimize the cost function. Based on the implementation of MATLAB program ,linear regression model and logistic regression classification model were analyzed. By comparing the convergence speed and complexity of the algorithm,the author obtained the application examples of different models. The result shows that choosing the better optimization algorithm according to the different data sets can make work faster.

关键词

梯度下降法/线性回归/逻辑回归/MATLAB

Key words

gradient descent method/linear regression/logistic regression/MATLAB

分类

数理科学

引用本文复制引用

孙娅楠,林文斌..梯度下降法在机器学习中的应用[J].苏州科技大学学报(自然科学版),2018,35(2):26-31,6.

基金项目

国家自然科学基金资助项目(11547311) (11547311)

苏州科技大学学报(自然科学版)

2096-3289

访问量0
|
下载量0
段落导航相关论文