solution = np.random.randint(0, 2, size=len(test_no_classes))
with open('my_solution.csv', 'w') as fout:
    print('Id', 'Prediction', sep=',', file=fout)
    for i, prediction in enumerate(solution):
        print(i, prediction, sep=',', file=fout)