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); |   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 | @end | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue
	
	Block a user