INNER CODE UNIT · Go

main

orneryd/NornicDB · cmd/kmeans-test-data/main.go:66

func main() {
	// Parse flags
	mode := flag.String("mode", "clusters", "Generation mode: synthetic, clusters, download, movies")
	count := flag.Int("count", 5000, "Number of embeddings to generate")
	dims := flag.Int("dims", 1024, "Embedding dimensions")
	numClusters := flag.Int("clusters", 20, "Number of natural clusters (for clusters mode)")
	outputDir := flag.String("output", "./data/kmeans-test", "Output directory")
	dbDir := flag.String("db", "", "NornicDB data directory (if set, imports directly)")
	dataset := flag.String("dataset", "", "Dataset: text-1024, movies-sample, movies-tmdb, movies-wiki (default depends on mode)")
	seed := flag.Int64("seed", 42, "Random seed for reproducibility")
	flag.Parse()

	rand.Seed(*seed)

	// Set default dataset based on mode
	effectiveDataset := *dataset
	if effectiveDataset == "" {
		switch *mode {

View source record →