Spaces:

snaramirez872
/

Finetuning-Toxicity-Model

Runtime error

snaramirez872 commited on Apr 26, 2023

Commit

a6ebfc7

1 Parent(s): f74bf51

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ class MultiLabelDataset(set):
         }
 # Dataset and DataLoader
-trainSize = 0.8
 trainData=new.sample(frac=trainSize,random_state=200)
 testData=new.drop(trainData.index).reset_index(drop=True)
 trainData = trainData.reset_index(drop=True)

         }
 # Dataset and DataLoader
+trainSize = 0.4
 trainData=new.sample(frac=trainSize,random_state=200)
 testData=new.drop(trainData.index).reset_index(drop=True)
 trainData = trainData.reset_index(drop=True)