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 中繼資料。