badjware
/
the-algorithm
mirror of https://github.com/twitter/the-algorithm.git
1
0
Fork 0
Code Releases Activity

Update trust_and_safety_models/toxicity/train.py

This commit is contained in:
Harshil 2023-04-02 00:56:46 +05:30 committed by GitHub
parent c94f4a42b9
commit 3051db2629
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
trust_and_safety_models/toxicity/train.py
View File

@ -387,7 +387,7 @@ class Trainer(object):
fold=i,
)
else:
a = input("Are you sure you want to do multiple fold training? (y/n)")
a = input("Are you sure you want to do multiple fold training? (y/N)")
if a.lower() == "y":
for mb_generator, steps_per_epoch, val_data, test_data in self.mb_loader(full_df=df):
self._train_single_fold(