moeen-m commited on
Commit
d7a6577
·
1 Parent(s): 30e1da9

Update predefined.py

Browse files
Files changed (1) hide show
  1. predefined.py +3 -3
predefined.py CHANGED
@@ -488,19 +488,19 @@ def get_output_new(w,wt,I_b=n_Identities,B_b=n_Behaviors,M_b=n_Modifiers,batch_s
488
  df_identity=pd.concat([Identities,new_df],axis=0)
489
  df2=out_df(data=q,predictions=preds,df_ident=df_identity)
490
  if cus_col:
491
- cols=[ 'EEA', 'EPA', 'EAA', 'ModA', 'Actor', 'Behavior', 'ModO', 'Object']
492
 
493
 
494
  if wt=='behavior':
495
  df_behavior=pd.concat([Behaviors,new_df],axis=0)
496
  df2=out_df(data=q,predictions=preds,df_beh=df_behavior)
497
  if cus_col:
498
- cols=['EEB', 'EPB', 'EAB', 'ModA', 'Actor', 'Behavior', 'ModO', 'Object']
499
  if wt=='modifier':
500
  df_modifier=pd.concat([Modifiers,new_df],axis=0)
501
  df2=out_df(data=q,predictions=preds,df_mod=df_modifier)
502
  if cus_col:
503
- cols=['EEMA', 'EPMA', 'EAMA', 'ModA', 'Actor', 'Behavior', 'ModO', 'Object']
504
  df=pd.concat([df,df2],axis=0)
505
  return(df[columnss])
506
 
 
488
  df_identity=pd.concat([Identities,new_df],axis=0)
489
  df2=out_df(data=q,predictions=preds,df_ident=df_identity)
490
  if cus_col:
491
+ columnss=[ 'EEA', 'EPA', 'EAA', 'ModA', 'Actor', 'Behavior', 'ModO', 'Object']
492
 
493
 
494
  if wt=='behavior':
495
  df_behavior=pd.concat([Behaviors,new_df],axis=0)
496
  df2=out_df(data=q,predictions=preds,df_beh=df_behavior)
497
  if cus_col:
498
+ columnss=['EEB', 'EPB', 'EAB', 'ModA', 'Actor', 'Behavior', 'ModO', 'Object']
499
  if wt=='modifier':
500
  df_modifier=pd.concat([Modifiers,new_df],axis=0)
501
  df2=out_df(data=q,predictions=preds,df_mod=df_modifier)
502
  if cus_col:
503
+ columnss=['EEMA', 'EPMA', 'EAMA', 'ModA', 'Actor', 'Behavior', 'ModO', 'Object']
504
  df=pd.concat([df,df2],axis=0)
505
  return(df[columnss])
506