AutoKeras:基於 Keras 的 AutoML 系統。它由德克薩斯農工大學的 DATA 實驗室開發。AutoKeras 的目標是讓每個人都可以使用機器學習。
AutoKeras 的社群相當活躍,您可以找到許多解決方案。
AutoKeras 目前支持的 Auto MLs 包含:
ImageClassifier
ImageRegressor
TextClassifier
TextRegressor
StructuredDataClassifier
StructuredDataRegressor
TimeseriesForecaster
上述官方範例文件及對應可執行的 Colab ,另外文件也指出,如果需要更加進階、客製自動搜尋模型,可以用AutoModel
以及其參數實現,歡迎您嘗試。
以下以可以發現與一般訓練流程差不多,以下以官方範例Image Classification 說明, 官方 Colab 範例 供參考,但提醒您會跑超過半小時,您要測試時建議減少 epochs 等參數。
安裝 autokeras
模組。
!pip install autokeras
訓練模型(摘述),與 tf.keras
訓練方式類似。
import autokeras as ak
# 初始化實例
clf = ak.ImageClassifier(overwrite=True, max_trials=1)
# 訓練模型,會比較花時間,跑著跑著最佳模型與參數就出來了。
clf.fit(x_train, y_train, epochs=10)
最終結果
您可以注意到,最佳模型結果已經儲存在 ./image_classifier/best_model/
之中,包含save_model.pb
與 keras_metadata.pb
中繼資料。