#! /usr/bin/env bash

function fashion_mnist() {
    local task=$(abcli_unpack_keyword $1 help)

    if [ $task == "help" ] ; then
        abcli_help_line "fashion_mnist ingest" \
            "ingest fashion_mnist data."
        abcli_help_line "fashion_mnist predict object_1" \
            "run fashion_mnist model object_1 predict."
        abcli_help_line "fashion_mnist train" \
            "train fashion_mnist."

        if [ "$(abcli_keyword_is $2 verbose)" == true ] ; then
            python3 -m fashion_mnist --help
        fi

        return
    fi

    if [ "$task" == "ingest" ] ; then
        python3 -m fashion_mnist \
            ingest \
            --output_path $abcli_object_path \
            ${@:2}
        return
    fi

    if [ "$task" == "predict" ] ; then
        abcli_fashion_mnist ingest
        abcli_image_classifier_predict ${@:2}
    fi

    if [ "$task" == "train" ] ; then
        local model_object=$abcli_object_name

        abcli_select - ~trail
        abcli_fashion_mnist ingest
        local data_object=$abcli_object_name

        abcli_select $model_object ~trail
        abcli_image_classifier_train \
            $data_object \
            ~color,~convnet \
            ${@:2}

        return
    fi

    abcli_log_error "-fashion_mnist: $task: command not found."
}

function abcli_fashion_mnist() {
    fashion_mnist $@
}