test

.idea/workspace.xml

@@ -2,7 +2,6 @@
 <project version="4">
   <component name="ChangeListManager">
     <list default="true" id="d25a65da-2ba0-4272-a0a5-c59cbecb6088" name="Default Changelist" comment="">
-      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/dev-0/out.tsv" beforeDir="false" afterPath="$PROJECT_DIR$/dev-0/out.tsv" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/dev-0/out_float.tsv" beforeDir="false" afterPath="$PROJECT_DIR$/dev-0/out_float.tsv" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/s.py" beforeDir="false" afterPath="$PROJECT_DIR$/s.py" afterDir="false" />

s.py

@@ -177,7 +177,7 @@ testA_x = torch.cat([testA_x_temp2, testA_x_words_onehot], 1)
 dataset_train = TrainDataset(x, y)
 trainloader=DataLoader(dataset=dataset_train, batch_size=minibatch_size, shuffle=True)
 
-def train_loop(i = 100):
+def train_loop(i = 50):
     for i in range(i):
         for xb, yb_expected in trainloader:
             optimizer.zero_grad()
@@ -197,13 +197,13 @@ def train_loop(i = 100):
             dev_y_pred_float_df = pandas.DataFrame(dev_y_pred_float_tensor.detach().numpy())
             auc_score = roc_auc_score(dev_y_test, dev_y_pred_float_df)
             print("auc:\t", auc_score, "\tloss:\t", loss.item())
-            if ((auc_score > 0.80)):
+            if ((auc_score > 0.70)):
                 break
 
             loss.backward()
             optimizer.step()
 
-        if ((auc_score > 0.80)):
+        if ((auc_score > 0.70)):
             break
         #print(loss)