例えば、mnistのような画像データを扱うとします。
ロードしたデータが
print(all_data.shape)
=> (3000,1,28,28)
となっており、3000枚の画像データで、ピクセルが28×28というデータ情報となっています。
そこで、この画像データを20%の訓練データと80%のテストデータに分けたいです
train_data = []
test_data = all_data.copy()
#20%である600個のデータ数をint型でtに入れる
t = int(len(all_data_label)*0.2)
for u in range(t):
i = randint(0,len(all_data))
# i番目のデータを訓練データtdに要素を入れる train_data.append(all_data[i]) # テスト用データ作成のためにi番目のデータを削除する
*********************************************
このようなコードを考えました。そこで上のコードの[****************]部分にi番目のデータを削除するようにプログラムしたいのですが、うまくいきません。
・pop
・delete
・remove
などさまざまな手法を試しましたが、errorが出ました。
アドバイスをお願いします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。