@@ -30,8 +30,8 @@ def test_sst2_classical1_no_expose_val(monkeypatch):
3030 model4_loss , model4_acc = classify .main ()
3131 assert round (model4_loss ,2 ) >= 0.6
3232 assert round (model4_loss ,2 ) <= 0.7
33- assert round (model4_acc ,1 ) >= 0.5
34- assert round (model4_acc ,1 ) <= 0.65
33+ assert round (model4_acc ,2 ) >= 0.5
34+ assert round (model4_acc ,2 ) <= 0.65
3535
3636
3737 # test_sst2_classical1_yes_expose_val
@@ -53,8 +53,8 @@ def test_sst2_classical1_yes_expose_val(monkeypatch):
5353 model4_loss , model4_acc = classify .main ()
5454 assert round (model4_loss ,2 ) >= 0.6
5555 assert round (model4_loss ,2 ) <= 0.7
56- assert round (model4_acc ,2 ) >= 0.3
57- assert round (model4_acc ,2 ) <= 0.5
56+ assert round (model4_acc ,2 ) >= 0.4
57+ assert round (model4_acc ,2 ) <= 0.7
5858
5959
6060# def test_food_it_classical1(monkeypatch):
@@ -152,8 +152,8 @@ def test_uspantek_classical1_no_expose_val(monkeypatch):
152152 model4_loss , model4_acc = classify .main ()
153153 assert round (model4_loss ,2 ) >= 0.6
154154 assert round (model4_loss ,2 ) <= 0.75
155- assert round (model4_acc ,1 ) >= 0.4
156- assert round (model4_acc ,1 ) <= 0.5
155+ assert round (model4_acc ,2 ) >= 0.3
156+ assert round (model4_acc ,2 ) <= 0.5
157157
158158
159159
@@ -174,8 +174,8 @@ def test_uspantek_classical1_yes_expose_val(monkeypatch):
174174 model4_loss , model4_acc = classify .main ()
175175 assert round (model4_loss ,2 ) >= 0.6
176176 assert round (model4_loss ,2 ) <= 0.7
177- assert round (model4_acc ,1 ) >= 0.4
178- assert round (model4_acc ,1 ) <= 0.6
177+ assert round (model4_acc ,2 ) >= 0.4
178+ assert round (model4_acc ,2 ) <= 0.6
179179
180180
181181# python classify.py --dataset uspantek --parser BobCatParser --ansatz SpiderAnsatz --model14type PytorchModel
@@ -198,15 +198,11 @@ def test_uspantek_classical2_no_expose_val(monkeypatch):
198198 model4_loss , model4_acc = classify .main ()
199199 assert round (model4_loss ,2 ) >= 0.68
200200 assert round (model4_loss ,2 ) <= 0.75
201- assert round (model4_acc ,1 ) >= 0.5
202- assert round (model4_acc ,1 ) <= 0.6
201+ assert round (model4_acc ,2 ) >= 0.5
202+ assert round (model4_acc ,2 ) <= 0.7
203203
204204
205- # python classify.py --dataset uspantek --parser BobCatParser
206- # --ansatz SpiderAnsatz --model14type PytorchModel --trainer PytorchTrainer
207- # --epochs_train_model1 7 --no_of_training_data_points_to_use 20
208- # --no_of_val_data_points_to_use 10 --max_tokens_per_sent 10
209- # --expose_model1_val_during_model_initialization
205+ # python classify.py --dataset uspantek --parser BobCatParser --ansatz SpiderAnsatz --model14type PytorchModel --trainer PytorchTrainer --epochs_train_model1 7 --no_of_training_data_points_to_use 20 --no_of_val_data_points_to_use 10 --max_tokens_per_sent 10 --expose_model1_val_during_model_initialization
210206
211207
212208def test_uspantek_classical2_yes_expose_val (monkeypatch ):
@@ -224,7 +220,7 @@ def test_uspantek_classical2_yes_expose_val(monkeypatch):
224220 '--expose_model1_val_during_model_initialization'
225221 ])
226222 model4_loss , model4_acc = classify .main ()
227- assert round (model4_loss ,2 ) >= 0.68
223+ assert round (model4_loss ,2 ) >= 0.6
228224 assert round (model4_loss ,2 ) <= 0.75
229- assert round (model4_acc ,2 ) >= 0.49
225+ assert round (model4_acc ,2 ) >= 0.3
230226 assert round (model4_acc ,2 ) <= 0.6
0 commit comments