update docs for write comprehension eval

quantling · Jun 20, 2021 · af2565b · af2565b
1 parent 38526d4
commit af2565b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/docs/src/man/output.md b/docs/src/man/output.md
@@ -7,12 +7,15 @@ CurrentModule = JudiLing
 ```@docs
     write2csv
     write2df
+    write_comprehension_eval
     write2csv(res, data, cue_obj_train, cue_obj_val, filename)
     write2csv(gpi::Vector{Gold_Path_Info_Struct}, filename)
     write2csv(ts::Threshold_Stat_Struct, filename)
     write2df(res, data, cue_obj_train, cue_obj_val)
     write2df(gpi::Vector{Gold_Path_Info_Struct})
     write2df(ts::Threshold_Stat_Struct)
+    write_comprehension_eval(SChat, SC, data, target_col, filename)
+    write_comprehension_eval(SChat, SC, SC_rest, data, data_rest, target_col, filename)
     save_L_matrix(L, filename)
     load_L_matrix(filename)
     save_S_matrix(S, filename, data, target_col)

diff --git a/src/output.jl b/src/output.jl
@@ -533,7 +533,7 @@ function load_S_matrix(filename; header = false, sep = " ")
 end
 
 """
-    write_comprehension_eval(SChat, SC, data, filename)
+    write_comprehension_eval(SChat, SC, data, target_col, filename)
 
 Write comprehension evaluation into a CSV file, include target and predicted 
 ids and indentifiers and their correlations.
@@ -599,7 +599,7 @@ function write_comprehension_eval(
 end
 
 """
-    write_comprehension_eval(SChat, SC, data, filename)
+    write_comprehension_eval(SChat, SC, SC_rest, data, data_rest, target_col, filename)
 
 Write comprehension evaluation into a CSV file for both training and validation
 datasets, include target and predicted ids and indentifiers and their