Added iOS test for different themes in text embedder
This commit is contained in:
parent
a512e6b5f5
commit
b5b10e7681
|
@ -139,4 +139,31 @@ static const float kDoubleDiffTolerance = 1e-4;
|
|||
XCTAssertEqualWithAccuracy(cosineSimilarity.doubleValue, 0.999937f, kDoubleDiffTolerance);
|
||||
}
|
||||
|
||||
- (void)testEmbedWithBertAndDifferentThemesSucceeds {
|
||||
MPPTextEmbedder *textEmbedder =
|
||||
[self textEmbedderFromModelFileWithName:kBertTextEmbedderModelName];
|
||||
|
||||
MPPEmbedding *embedding1 =
|
||||
[self assertFloatEmbeddingResultsOfEmbedText:
|
||||
@"When you go to this restaurant, they hold the pancake upside-down before they "
|
||||
@"hand it to you. It's a great gimmick."
|
||||
usingTextEmbedder:textEmbedder
|
||||
hasCount:512
|
||||
firstValue:42.0832];
|
||||
|
||||
MPPEmbedding *embedding2 =
|
||||
[self assertFloatEmbeddingResultsOfEmbedText:
|
||||
@"Let's make a plan to steal the declaration of independence."
|
||||
usingTextEmbedder:textEmbedder
|
||||
hasCount:512
|
||||
firstValue:50.8856];
|
||||
|
||||
NSNumber *cosineSimilarity = [MPPTextEmbedder cosineSimilarityBetweenEmbedding1:embedding1
|
||||
andEmbedding2:embedding2
|
||||
error:nil];
|
||||
|
||||
// TODO: The similarity should likely be lower
|
||||
XCTAssertEqualWithAccuracy(cosineSimilarity.doubleValue, 0.963203f, kDoubleDiffTolerance);
|
||||
}
|
||||
|
||||
@end
|
||||
|
|
Loading…
Reference in New Issue
Block a user