dvc_session_test/src/logistic_regression.py

import plac
from sklearn.linear_model import LogisticRegression

from utils import evaluate_model, print_results, save_results, log_experiment, read_data, read_params


@plac.annotations(
    data_path=("Path to source data", "option", "i", str),
    out_path=("Path to save trained Model", "option", "o", str)
)
def main(data_path='data/features/', out_path='data/models/logistic/'):
    X_train, X_test, y_train, y_test = read_data(data_path)

    name = 'LogisticRegression'
    params = read_params('params.yaml', 'logistic')
    model = LogisticRegression(**params)
    model.fit(X_train, y_train)

    accuracy, c_matrix, fig = evaluate_model(model, X_test, y_test)
    print_results(accuracy, c_matrix, name)

    save_results(out_path, model, fig)
    log_experiment(out_path, params=params,
                   metrics=dict(accuracy=accuracy, confusion_matrics=c_matrix))


if __name__ == '__main__':
    plac.call(main)