機械学習(クラス分け)用のデータ作成処理

sample_write_test_data_1.py

import csv
import random

with open('test_data.csv', 'w') as f:
    writer = csv.writer(f, lineterminator='\n')
    writer.writerow(['Japanese', 'Mathematics', 'English', 'Evaluation'])
    for i in range(500):
        test = []
        for j in range(3):
            test.append(random.randint(0, 10) * 10)
        test_sum = 0
        for k in test:
            test_sum += k
        test.append((test_sum // 60) + 1)
        writer.writerow(test)

ブログの記事で作成しました。
興味のある方はご覧ください。

【機械学習】3教科のテスト結果を分類してみた➀
https://neko-py.com/machine-learning-classification-1