-class _kmeans_lsi_grid_sweep(grid_sweep):
- def __init__(self,
- inpath,
- outpath,
- lsi_dim,
- *args,
- **kwargs):
- self.lsi_dim = lsi_dim
- self.jobtype = kmeans_lsi_job
- super().__init__(self.jobtype, inpath, outpath, self.namer, self.lsi_dim, *args, **kwargs)
-
- def namer(self, *args, **kwargs):
- s = kmeans_grid_sweep.namer(self, *args[1:], **kwargs)
- s += f"_lsi-{self.lsi_dim}"
- return s
-
-class kmeans_lsi_grid_sweep(lsi_grid_sweep):
- def __init__(self,
- inpath,
- lsi_dims,
- outpath,
- n_clusters,
- n_inits,
- max_iters
- ):
-
- super().__init__(kmeans_lsi_job,
- _kmeans_lsi_grid_sweep,
- inpath,
- lsi_dims,
- outpath,
- n_clusters,
- n_inits,
- max_iters)
-