AutoKeras:用于深度学习的 AutoML 库

概述

一个基于 Keras 的 AutoML 系统。它由德克萨斯 A&M 大学的数据实验室开发。AutoKeras 的目标是让每个人都能使用机器学习。

学习资源

一个简短的例子。

import autokeras as ak

clf = ak.ImageClassifier()
clf.fit(x_train, y_train)
results = clf.predict(x_test)

安装

[hidecontent type="logged" desc="隐藏内容：登录后可查看"]

要安装软件包，请pip按如下方式使用安装：

pip3 install autokeras

详细安装步骤

要求

Python 3：按照 TensorFlow 安装步骤安装 Python 3。 Pip：按照 TensorFlow 安装步骤安装 Pip。 Tensorflow >= 2.3.0：AutoKeras 基于 TensorFlow。请按照本教程为 python3 安装 TensorFlow。 GPU 设置（可选）：如果您的机器上有 GPU 并想使用它们来加速训练，您可以按照本教程进行设置。

安装 AutoKeras

AutoKeras 仅支持Python 3。如果您按照前面的步骤使用 virtualenv 安装 tensorflow，则只需激活 virtualenv 并使用以下命令安装 AutoKeras。

pip install git+https://github.com/keras-team/keras-tuner.git
pip install autokeras

如果您没有使用 virtualenv，而是使用python3命令执行您的 python 程序，请使用以下命令安装 AutoKeras。

python3 -m pip install git+https://github.com/keras-team/keras-tuner.git
python3 -m pip install autokeras

如您想详细了解这个项目，请按照安装指南了解更多详细信息。

注意：目前，AutoKeras 仅与Python >= 3.7和TensorFlow >= 2.8.0兼容。

[/hidecontent]