mirror of
https://github.com/hanzoai/anomalib.git
synced 2026-10-30 18:48:11 +00:00
zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
32 lines
1.1 KiB
Plaintext
32 lines
1.1 KiB
Plaintext
class TrainJob(Job):
|
|
name = "train"
|
|
|
|
def __init__(self, lr: float, backbone: str, stride: int):
|
|
self.lr = lr
|
|
self.backbone = backbone
|
|
self.stride = stride
|
|
|
|
def run(self, task_id: int | None = None) -> dict:
|
|
print(f"Training with lr: {self.lr}, backbone: {self.backbone}, stride: {self.stride}")
|
|
time.sleep(2)
|
|
score = np.random.uniform(0.7, 1.0)
|
|
return {"lr": self.lr, "backbone": self.backbone, "stride": self.stride, "score": score}
|
|
|
|
@staticmethod
|
|
def collect(results: list[dict]) -> pd.DataFrame:
|
|
"""Collect all individual runs into a dict of lists."""
|
|
output: dict = {}
|
|
for key in results[0]:
|
|
output[key] = []
|
|
for result in results:
|
|
for key, value in result.items():
|
|
output[key].append(value)
|
|
return pd.DataFrame(output)
|
|
|
|
@staticmethod
|
|
def save(results: pd.DataFrame) -> None:
|
|
"""Save results in a csv file."""
|
|
file_path = Path("runs") / TrainJob.name
|
|
file_path.mkdir(parents=True, exist_ok=True)
|
|
results.to_csv(file_path / "results.csv", index=False)
|