​

GridSearch Class

The GridSearch class is used for performing grid search on a given chain.

​

Attributes

• experiments: A list of Experiment objects.
• experiment_results: A DataFrame containing the results of the experiments.
• console: A Rich Console object.
• vectorstore_list: A list of vector stores.
• retrievers_list: A list of retrievers.
• chain: A Chain object.

​

Methods

​

__init__(self, chain)

Initializes a GridSearch object.

​

Parameters

• chain (Chain): A Chain object.

grid_search = GridSearch(chain=RetrievalQA)

​

create_experiments(self, param_grid, loader=None, documents=None)

Creates a list of Experiment objects based on the given parameter grid.

​

Parameters

• param_grid (dict): A dictionary containing the parameter grid.
• loader (Loader): A Loader object.
• documents (list): A list of documents.

​

Example