taratrankennedy commited on
Commit
fbcb1fd
·
verified ·
1 Parent(s): 05b9388

Printing model_evaluation to the terminal

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -3,6 +3,8 @@ from sentence_transformers import SentenceTransformer, util
3
  import openai
4
  import os
5
  from model_evaluation import process_dataset
 
 
6
 
7
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
8
 
@@ -145,5 +147,16 @@ if __name__ == "__main__":
145
  process_dataset("val_dataset.csv", "updated_val_dataset.csv")
146
  demo.launch(share=True)
147
 
 
 
 
 
 
 
 
 
 
 
 
148
 
149
  demo.launch(share=True)
 
3
  import openai
4
  import os
5
  from model_evaluation import process_dataset
6
+ import pandas as pd
7
+
8
 
9
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
10
 
 
147
  process_dataset("val_dataset.csv", "updated_val_dataset.csv")
148
  demo.launch(share=True)
149
 
150
+ # The following are for model_evaluation.py
151
+ # Load the results
152
+ train_results = pd.read_csv("updated_train_dataset.csv")
153
+ val_results = pd.read_csv("updated_val_dataset.csv")
154
+
155
+ # Display the first few rows of each dataset
156
+ print("Train Dataset Results:")
157
+ print(train_results.head())
158
+
159
+ print("\nValidation Dataset Results:")
160
+ print(val_results.head())
161
 
162
  demo.launch(share=True)