Skip to content

E
ease
Commit 3df28416 by Vik Paruchuri
Options

Edit test thresholds

parent c5eae177
Showing with 3 additions and 2 deletions
ease/tests/test_model_accuracy.py
...@@ -10,7 +10,7 @@ ROOT_PATH = os.path.abspath(__file__) ...@@ -10,7 +10,7 @@ ROOT_PATH = os.path.abspath(__file__)
TEST_PATH = os.path.abspath(os.path.join(ROOT_PATH, "..")) TEST_PATH = os.path.abspath(os.path.join(ROOT_PATH, ".."))
CHARACTER_LIMIT = 1000 CHARACTER_LIMIT = 1000
TRAINING_LIMIT = 40 TRAINING_LIMIT = 100
QUICK_TEST_LIMIT = 5 QUICK_TEST_LIMIT = 5
class DataLoader(): class DataLoader():
...@@ -112,6 +112,7 @@ class GenericTest(object): ...@@ -112,6 +112,7 @@ class GenericTest(object):
self.assertTrue(results['success']) self.assertTrue(results['success'])
def test_scoring_accuracy(self): def test_scoring_accuracy(self):
random.seed(1)
model_creator = ModelCreator(self.scores, self.text) model_creator = ModelCreator(self.scores, self.text)
results = model_creator.create_model() results = model_creator.create_model()
self.assertTrue(results['success']) self.assertTrue(results['success'])
...@@ -127,7 +128,7 @@ class PolarityTest(unittest.TestCase,GenericTest): ...@@ -127,7 +128,7 @@ class PolarityTest(unittest.TestCase,GenericTest):
#These will increase if we allow more data in. #These will increase if we allow more data in.
#I am setting the amount of data low to allow tests to finish quickly (40 training essays, 1000 character max for each) #I am setting the amount of data low to allow tests to finish quickly (40 training essays, 1000 character max for each)
expected_kappa_min = .26 expected_kappa_min = .26
expected_mae_max = .15 expected_mae_max = .2
def setUp(self): def setUp(self):
self.generic_setup() self.generic_setup()
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