| # Initialize ErnieBot Deep Search | |
| search_engine = ErnieBotDeepSearch() | |
| # Example search | |
| query = "深度学习的应用" | |
| documents = [ | |
| "机器学习是人工智能的一个分支", | |
| "深度学习在图像识别领域取得了突破性进展", | |
| "自然语言处理是深度学习的重要应用领域" | |
| ] | |
| # Perform deep search | |
| results = search_engine.deep_search(query, documents) | |
| ranked_results = torch.argsort(results, descending=True) | |
| # Training example | |
| training_batch = ( | |
| query, | |
| [documents[1], documents[2]], # positive examples | |
| [documents[0]] # negative examples | |
| ) | |
| loss = search_engine.train_step(training_batch) | |