机器学习笔记 - 使用Deeplearning4j和浅层神经网络进行分类

坐望云起

已于 2022-04-09 10:16:04 修改

阅读量1.2k

点赞数

分类专栏：深度学习从入门到精通文章标签： deeplearning4j 鸢尾花神经网络机器学习 java

于 2022-04-09 10:15:53 首次发布

本文链接：https://blog.youkuaiyun.com/bashendixie5/article/details/124055745

版权

深度学习从入门到精通专栏收录该内容

798 篇文章 ¥99.90 ¥299.90

订阅专栏

这篇博客详细介绍了如何利用 Deeplearning4j 框架创建一个简单的神经网络，以对鸢尾花数据集进行分类。博主首先介绍环境设置，包括 Maven 的依赖配置。接着，数据准备部分涉及数据读取、打乱、标准化和拆分。在模型配置中，博主选择了双曲正切激活函数、高斯权重初始化和L2正则化。经过训练后，博主保存了模型，并展示了训练过程的日志，显示模型在鸢尾花分类任务上表现出较高的准确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、环境设置

我们使用deeplearning4j (dl4j) 库创建一个简单的神经网络，该库是一种现代且强大的机器学习工具。

让我们将所需的库添加到我们的 Maven pom.xml文件中，如下配置。

<properties>
    <dl4j-master.version>1.0.0-M2</dl4j-master.version>
    <java.version>1.8</java.version>
</properties>

<dependencies>
		<!-- deeplearning4j-core: contains main functionality and neural networks -->
		<dependency>
			<groupId>org.deeplearning4j</groupId>
			<artifactId>deeplearning4j-core</artifactId>
			<version>${dl4j-master.version}</version>
		</dependency>
		<dependency>
			<groupId>org.nd4j</groupId>
			<artifactId>nd4j-native</artifactId>
			<version>${dl4j-master.versi

了解本专栏