Spaces:

jaothan
/

GenAI_project

Sleeping

App Files Files Community

GenAI_project / tests /test_metrics.py

jaothan

Upload 24 files

fa64206 verified almost 2 years ago

Raw History Blame Contribute Delete

407 Bytes

import unittest
from evaluations.evaluate_models import evaluate_model

class TestMetrics(unittest.TestCase):

    def test_evaluate_model(self):
        model = train_full_finetune_model(train_data, val_data)
        results = evaluate_model(model, val_data)
        self.assertIn('accuracy', results)
        self.assertIn('f1_score', results)

if __name__ == '__main__':
    unittest.main()

	import unittest
	from evaluations.evaluate_models import evaluate_model

	class TestMetrics(unittest.TestCase):

	def test_evaluate_model(self):
	model = train_full_finetune_model(train_data, val_data)
	results = evaluate_model(model, val_data)
	self.assertIn('accuracy', results)
	self.assertIn('f1_score', results)

	if __name__ == '__main__':
	unittest.main()