Spaces:
Running
Running
Sadjad Alikhani
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -378,7 +378,7 @@ def identical_train_test_split(output_emb, output_raw, labels, train_percentage)
|
|
378 |
indices = torch.randperm(N) # Randomly shuffle the indices
|
379 |
|
380 |
# Calculate the split index for test (10% of the data)
|
381 |
-
test_split_index = int(N * 0.
|
382 |
|
383 |
# Test indices (first 10% of the data)
|
384 |
test_indices = indices[:test_split_index]
|
|
|
378 |
indices = torch.randperm(N) # Randomly shuffle the indices
|
379 |
|
380 |
# Calculate the split index for test (10% of the data)
|
381 |
+
test_split_index = int(N * 0.20)
|
382 |
|
383 |
# Test indices (first 10% of the data)
|
384 |
test_indices = indices[:test_split_index]
|