Tuberrr
/
JCC-DeepOD
forked from JointCloud/JCC-DeepOD
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #39 from crishna0401/main 

Update utils.py
This commit is contained in:
Hongzuo Xu 2023-12-11 09:35:11 +08:00 committed by GitHub
parent f790ee5dec d844e0ca17
commit a1ec53af7e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
testbed/utils.py
View File

@ -142,7 +142,7 @@ def read_data(file, split='50%-normal', normalization='z-score', seed=42):
f'After splitting: training/testing [{len(x_train)}/{len(x_test)}]')
elif split == '60%':
x_train, y_train, x_test, y_test = train_test_split(x, y, shuffle=True, random_state=seed,
x_train, x_test, y_train, y_test = train_test_split(x, y, shuffle=True, random_state=seed,
test_size=0.4, stratify=y)
else: