Skip to content

Packaged inference model - #149

Merged
vdplasthijs merged 8 commits into
developfrom
feature/packaged_inference
Aug 31, 2026
Merged

Packaged inference model#149
vdplasthijs merged 8 commits into
developfrom
feature/packaged_inference

Conversation

@gabrieletijunaityte

@gabrieletijunaityte gabrieletijunaityte commented Aug 27, 2026

Copy link
Copy Markdown
Contributor

What does this PR do?

Packaged inference model for NILU, open for changes based on their needs.

  • This load model from inference checkpoint
  • Encods texts from user input, predefined list, or txt file.
  • Intakes geo_data.csv (same as input data for the alignment model) with name_loc and emb_1 ... columns
  • Embed this geo_data and makes predictions.
  • Saves results or prints them.

Before submitting

  • Did you make sure title is self-explanatory and the description concisely explains the PR?
  • Did you make sure your PR does only one thing, instead of bundling different changes together?
  • Did you list all the breaking changes introduced by this pull request?
  • Did you test your PR locally with pytest command?

@gabrieletijunaityte
gabrieletijunaityte changed the base branch from main to develop August 27, 2026 15:03
@gabrieletijunaityte gabrieletijunaityte changed the title Feature/packaged inference Packaged inference model Aug 27, 2026

@vdplasthijs vdplasthijs left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ok all looks great, just 1 minor comment, see below!

Comment thread src/models/inference_model.py Outdated
@gabrieletijunaityte

Copy link
Copy Markdown
Contributor Author

I fixed the import issue!

Also now created an inference directory with UC subfolder inference/s2bms/ here we put data for inference and in version folders like v1 we place checkpoints. Im preparing instructions too!

@vdplasthijs
vdplasthijs merged commit d35918f into develop Aug 31, 2026
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants