Commit f22bbf0c authored by Joel Oksanen's avatar Joel Oksanen
Browse files

Added test path for amazon reviews

parent b74782c2
...@@ -10,6 +10,7 @@ from sklearn import metrics ...@@ -10,6 +10,7 @@ from sklearn import metrics
semeval_2014_train_path = 'data/SemEval-2014/Laptop_Train_v2.xml' semeval_2014_train_path = 'data/SemEval-2014/Laptop_Train_v2.xml'
semeval_2014_test_path = 'data/SemEval-2014/Laptops_Test_Gold.xml' semeval_2014_test_path = 'data/SemEval-2014/Laptops_Test_Gold.xml'
amazon_test_path = 'data/Amazon/amazon_camera_test.xml'
trained_model_path = 'semeval_2014.pt' trained_model_path = 'semeval_2014.pt'
BATCH_SIZE = 32 BATCH_SIZE = 32
...@@ -93,4 +94,4 @@ class BertAnalyzer: ...@@ -93,4 +94,4 @@ class BertAnalyzer:
sentiment_analyzer = BertAnalyzer() sentiment_analyzer = BertAnalyzer()
sentiment_analyzer.load_saved() sentiment_analyzer.load_saved()
sentiment_analyzer.evaluate(semeval_2014_test_path) sentiment_analyzer.evaluate(amazon_test_path)
\ No newline at end of file \ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment