forked from enviPath/enviPy
Merge remote-tracking branch 'origin/develop' into enhancement/dataset
# Conflicts: # epdb/models.py # tests/test_enviformer.py # tests/test_model.py
This commit is contained in:
@ -3,7 +3,7 @@ from datetime import datetime
|
||||
from tempfile import TemporaryDirectory
|
||||
from django.test import TestCase, tag
|
||||
from epdb.logic import PackageManager
|
||||
from epdb.models import User, EnviFormer, Package, Setting, Pathway
|
||||
from epdb.models import User, EnviFormer, Package, Setting
|
||||
from epdb.tasks import predict_simple, predict
|
||||
|
||||
|
||||
@ -48,9 +48,7 @@ class EnviFormerTest(TestCase):
|
||||
|
||||
mod.build_dataset()
|
||||
mod.build_model()
|
||||
mod.multigen_eval = True
|
||||
mod.save()
|
||||
mod.evaluate_model(n_splits=2)
|
||||
mod.evaluate_model(True, eval_packages_objs, n_splits=2)
|
||||
|
||||
mod.predict("CCN(CC)C(=O)C1=CC(=CC=C1)C")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user