Update predefined.py
Browse files- 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 |
-
|
| 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 |
-
|
| 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 |
-
|
| 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 |
|