Function save_results

Logistic-Regression/classifier_corrected.py:42–47

Given a vector of predictions, save results in CSV format.

(predictions, filename)

def save_results(predictions, filename):
    """Given a vector of predictions, save results in CSV format."""
    with open(filename, 'w') as f:
        f.write("id,ACTION\n")
        for i, pred in enumerate(predictions):
            f.write("%d,%f\n" % (i + 1, pred))


def main():