add notebook

Tikquuss · Aug 19, 2020 · e526a78 · e526a78
1 parent ac2ae1a
commit e526a78
Show file tree

Hide file tree

Showing 3 changed files with 341 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -29,5 +29,5 @@ path_to_eula=my/eula.txt,my/eula.md,my/eula.pdf,my/eula.docx
 python eulascript/eula.py --model_folder $model_folder --path_to_eula $path_to_eula --output_dir $output_dir
 ```
 
-**Note**: the [samples](samples) folder contains some user licenses and a [notebook](samples/notebook.ipynb) illustrating 
+**Note**: the [samples](samples) folder contains some user licenses and a [notebook](notebook.ipynb) illustrating all. 
 
diff --git a/eula.py b/eula.py
@@ -32,6 +32,8 @@ def main(params):
     predict_method = get_ktrain_predict_method(ktrain_predictor = ktrain.load_predictor(params.model_folder))      
 
     for eula_file in params.eula_files :
+
+        print("============ eula_file : ", eula_file, " ============")
 
         clause_list = get_content(eula_file)
 
@@ -49,6 +51,8 @@ def main(params):
                 i += 1
             csv_file = file_name+'.'+str(i)+'.csv'
 
+        print("============ csv_file : ", csv_file, " ============")
+
         pd.DataFrame(zip(clause_list, labels, probabilities)).to_csv(csv_file, header= ["clauses", "labels", "probabilities"])
 
 if __name__ == '__main__':