Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2025-02-15 11:49:14 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge 3051db26297cac671994462aea9fcbd35e364103 into fb54d8b54984f89f7dba90a18e7c3048421464c3

Browse Source
This commit is contained in:
Harshil 2023-05-22 17:38:15 -05:00 committed by GitHub
commit f218429159
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
trust_and_safety_models/toxicity/train.py
View File

@ -387,15 +387,16 @@ class Trainer(object):
fold=i,
)
else:
raise ValueError("Sure you want to do multiple fold training")
for mb_generator, steps_per_epoch, val_data, test_data in self.mb_loader(full_df=df):
self._train_single_fold(
mb_generator=mb_generator,
val_data=val_data,
test_data=test_data,
steps_per_epoch=steps_per_epoch,
fold=i,
)
i += 1
if i == 3:
break
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(
mb_generator=mb_generator,
val_data=val_data,
test_data=test_data,
steps_per_epoch=steps_per_epoch,
fold=i,
)
i += 1
if i == 3:
break