renaming all files - moving training to be a single file for transfer vs not transfer learning. Made the testing file test all models. Needs to be updated to only update with new models.

This commit is contained in:
Lucas
2022-06-01 17:46:55 -04:00
parent ab0b7a0a4a
commit 1b539d6945
960 changed files with 338 additions and 1411 deletions
+8
View File
@@ -0,0 +1,8 @@
from collections import Callable
class ModelWrapper(object):
def __init__(self, model_func:Callable, model_preprocessor:Callable, name:str):
self.model_func = model_func
self.model_preprocessor = model_preprocessor
self.name = name