55e6def4d984c34679ab3c48a90055e93d351f79,feat_extract.py,,,#,89

Before Change

    one_hot_encode[np.arange(n_labels), labels] = 1
    return one_hot_encode

if __name__ == "__main__":
    // Get features and labels
    r = os.listdir("data/")
    r.sort()
    features, labels = parse_audio_files("data", r)
    np.save("feat.npy", features)
    np.save("label.npy", labels)

    // Predict new
    r = os.listdir("predict/")
    r.sort()
    features, filenames = parse_predict_files("predict")
    np.save("predict_feat.npy", features)
    np.save("predict_filenames.npy", filenames)

After Change

    np.save("predict_feat.npy", features)
    np.save("predict_filenames.npy", filenames)

if __name__ == "__main__": main()

Instances