update

rubix-coder · rubix-coder · commit acdc540d86c7 · 2025-05-25T18:21:55.000+05:30
diff --git a/leetcode-master/scratch_codes/pytorch_huggingface_tutorial/slice_dice_dataset.py b/leetcode-master/scratch_codes/pytorch_huggingface_tutorial/slice_dice_dataset.py
@@ -36,3 +36,18 @@ def filter_nones(x):
 # Check that lowercasing worked
 
 print(f'check if lower casing worked -> {drug_dataset["train"]["condition"][:3]}')
+
+# Creating new columns
+
+def compute_review_length(example):
+    return {"review_length": len(example["review"].split())}
+
+
+drug_dataset = drug_dataset.map(compute_review_length)
+# Inspect the first training example
+print(f"Inspect the first training example -> {drug_dataset['train'][0]}")
+print(f"sort the dataset with the new column -> {drug_dataset["train"].sort("review_length")[:3]}")
+
+# Datasets + Dataframe = Pandas <B
+
+
