Upload predefined_bertnn.py

predefined_bertnn.py

@@ -638,9 +638,9 @@ def get_output_agg(w,wt,I_b=n_Identities,B_b=n_Behaviors,M_b=n_Modifiers,batch_s
       
     return(df)
 
-def ldr_new(I,B,M,N_df,WT,batch_size=32):
-    dt_ldr= [x for x in DataLoader([next(gen_new(I,B,M,N_df,WT)) for x in range(batch_size)],  batch_size=batch_size)][0]
-    return(dt_ldr)
+# def ldr_new(I,B,M,N_df,WT,batch_size=32):
+#     dt_ldr= [x for x in DataLoader([next(gen_new(I,B,M,N_df,WT)) for x in range(batch_size)],  batch_size=batch_size)][0]
+#     return(dt_ldr)