Skip to content

Conversation

@AndreasKarasenko
Copy link
Contributor

People should be able to use different distance metrics for the DynamicFewShotGPTClassifier. According to this SO answer we should use the same similarity metric that of the model. E.g. cosine for openai.
I simply propagated the metric variable up to DynamicFewShotGPTClassifier and tested it with the example from the documentation.

@AndreasKarasenko
Copy link
Contributor Author

I added a change for #90 in the second commit. From what I can tell the predict function is handled by the BaseClassifier mixin and we can simply cast to np.array.

@OKUA1 OKUA1 merged commit 951f861 into BeastByteAI:main May 17, 2024
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