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]
|