関数を使用してデータを、、、train_test_splitに分割し、それを CNN にフィードしていましたが、混同行列を作成して予測テンソルと比較するためにラベル テンソルにアクセスする方法がわからないという問題があります。私はそれが基本的な質問であることを知っていますが、とにかくあなたの助けに感謝します.X_trainX_testy_trainy_testutils.data.DataLoader
X_train, X_test, y_train, y_test = train_test_split(faces, emotions, test_size=0.1, random_state=42)
X_train, X_val, y_train, y_val = train_test_split(X_train, y_train, test_size=0.1, random_state=41)
そして私は使った
train = torch.utils.data.TensorDataset(torch.from_numpy(X_train), torch.from_numpy(y_train))
train_loader = torch.utils.data.DataLoader(train, batch_size=100, shuffle=True)
私のネットワークにデータを供給するために、train_setの後にtargets属性を入力するだけでラベルにアクセスできるようですが、そのようtrain_set.targetsには機能しません。ラベルを取得するにはどうすればよいですか?