Dylan Knutson 75e7a4538d Refactor Dockerfile and add fit_model functionality ...

- Updated the Dockerfile to rename the built binary from `mf-fitter` to `fit_model` for clarity.
- Introduced a new `fit_model_args.rs` file to define command-line arguments for the fit model process, including parameters for matrix factorization.
- Added `pg_types.rs` and `pgvector.rs` files to handle PostgreSQL type interactions and vector serialization/deserialization.
- Implemented the main logic for the fit model in `fit_model.rs`, including data loading, model training, and embedding saving.
- Enhanced `visualize_embeddings.rs` to load embeddings and clusters more efficiently.

These changes improve the organization and functionality of the model fitting process, making it more intuitive and maintainable.

2024-12-28 19:50:24 +00:00

1.3 KiB

Raw Permalink Blame History

View Raw