mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-06 11:37:44 +00:00
feat: add model removal and adjust some framework logic (#681)
* prune model task * add component_description * add model removal logic to component, hypo, and task gen * fix ci * adjust coder to meet the requirement of model removal * fix and refine the logic of model removal * add model removal logic in model_eval * fix ci * fix ci * prune some unnecessary codes
This commit is contained in:
@@ -9,28 +9,13 @@ class ModelTask(CoSTEERTask):
|
||||
self,
|
||||
name: str,
|
||||
description: str,
|
||||
architecture: str = "",
|
||||
*args,
|
||||
hyperparameters: Dict[str, str] = {},
|
||||
model_type: Optional[str] = None,
|
||||
**kwargs,
|
||||
) -> None:
|
||||
self.architecture: str = architecture
|
||||
self.hyperparameters: str = hyperparameters
|
||||
self.model_type: str | None = (
|
||||
model_type # Tabular for tabular model, TimesSeries for time series model, Graph for graph model, XGBoost for XGBoost model
|
||||
# TODO: More Models Supported
|
||||
)
|
||||
super().__init__(name=name, description=description, *args, **kwargs)
|
||||
|
||||
def get_task_information(self):
|
||||
task_desc = f"""name: {self.name}
|
||||
description: {self.description}
|
||||
"""
|
||||
if self.architecture:
|
||||
task_desc += f"architecture: {self.architecture}\n"
|
||||
if self.hyperparameters:
|
||||
task_desc += f"hyperparameters: {self.hyperparameters}\n"
|
||||
if self.model_type:
|
||||
task_desc += f"model_type: {self.model_type}\n"
|
||||
return task_desc
|
||||
|
||||
Reference in New Issue
Block a user