該項目自動選擇給定用例的最佳模型,提示或推理參數的過程,從而使您可以迭代其組合併視覺檢查結果。
它假設Ollama已安裝並在localhost或遠程服務器中提供終點。
這是一個對3種不同型號進行測試的簡單提示的實驗,看起來像:

(有關此工具輔助的評估過程的更深入的查看,請檢查https://dezoito.github.io/2023/12/12/27/rust-ollast-ollama-grid-search.html)。
檢查項目的發布頁面,或在側邊欄上。
從技術上講,術語“網格搜索”是指在一系列不同的模型HyperParams上進行迭代以優化模型性能,但這通常意味著諸如batch_size , learning_rate或number_of_epochs之類的參數,在訓練中更常用。
但是這裡的概念相似:
讓我們定義模型的選擇,提示和一些參數組合:

每個參數值將對每個選定模型中的每個參數值提交一次,生成一組響應。
同樣,您可以通過選擇不同的模型來執行A/B測試,並比較相同的提示/參數組合的結果,或在相似配置下測試不同的提示:

比較同一模型的不同提示的結果
您可以保存和管理您的提示(我們想使提示與Open WebUI兼容)

您可以通過鍵入“/”(也受開放webUI的啟發)來自動完成提示:

您可以列出,檢查或下載您的實驗:

對於明顯的錯誤和拼寫錯誤,請繼續提交PR。
如果您想提出一項新功能,更改現有功能或提出更複雜的內容,請在完成公關工作之前打開討論問題。
確保您已經安裝了生鏽。
克隆存儲庫(或叉子)
git clone https://github.com/dezoito/ollama-grid-search.git
cd ollama-grid-search安裝前端依賴項。
cd < project root >
# I'm using bun to manage dependencies,
# but feel free to use yarn or npm
bun install確保在檢查代碼時配置rust-analyzer以運行Clippy 。
如果您正在運行VS代碼,請將其添加到您的settings.json文件
{
...
"rust-analyzer.check.command" : " clippy " ,
}(或者,更好的是,只需使用與代碼中提供的設置文件)
在開發模式下運行應用程序
cd < project root > /
bun tauri dev去喝杯咖啡,因為這可能需要一段時間。
以下作品和論文引用了這個存儲庫:
Inouye,D&Lindo,L和Lee,R&Allen,E;計算機科學與工程高級論文:在Lora超參數上進行自動調節聖克拉拉大學,2024 https://scholarcommons.scu.edu/cgi/cgi/cgi/viewcontent.cgi?article=1271&context=cseng_seng_senior = cSeng_Senior = cseng_seniorior = cSeng_Seniorior = cSeng_Seniorior =
非常感謝@fabianlars, @peperroni21和@tomreidnz。