Pytorch delete features columns from dataset

Advertisements I have a dataset below and would like to delete features From A – F the dataset are converted from python dataframe dataset = datasets.DatasetDict({"train":Dataset.from_pandas(X_train), "test":Dataset.from_pandas(X_test), "val":Dataset.from_pandas(X_val), }) The dataset output like below DatasetDict({ train: Dataset({ features: [‘A’, ‘B’, ‘C’, ‘D’, ‘E’, ‘F’, ‘text’, ‘__index_level_0__’, ‘label’], num_rows: 1173 }) test: Dataset({ features: [‘A’, ‘B’,… Read More Pytorch delete features columns from dataset