日韩成人免费在线_国产成人一二_精品国产免费人成电影在线观..._日本一区二区三区久久久久久久久不

當前位置:首頁 > 科技  > 軟件

PyTorch 訓練,除了會訓練還要了解這些

來源: 責編: 時間:2024-09-10 09:48:10 126觀看
導讀讓我們討論一下在訓練過程中幫助你進行實驗的技術。我將提供一些理論、代碼片段和完整的流程示例。主要要點包括:數據集分割指標可重復性配置、日志記錄和可視化分割數據集我喜歡有訓練集、驗證集和測試集的分割。這里

讓我們討論一下在訓練過程中幫助你進行實驗的技術。我將提供一些理論、代碼片段和完整的流程示例。主要要點包括:4pC28資訊網——每日最新資訊28at.com

  • 數據集分割
  • 指標
  • 可重復性
  • 配置、日志記錄和可視化

4pC28資訊網——每日最新資訊28at.com

分割數據集

我喜歡有訓練集、驗證集和測試集的分割。這里沒什么好說的;你可以使用隨機分割,或者如果你有一個不平衡的數據集(就像在實際情況中經常發生的那樣)——分層分割。4pC28資訊網——每日最新資訊28at.com

對于測試集,嘗試手動挑選一個“黃金數據集”,包含你希望模型擅長的所有示例。測試集應該在實驗之間保持不變。它應該只在你完成模型訓練后使用。這將在部署到生產環境之前給你客觀的指標。別忘了,你的數據集應該盡可能接近生產環境,這樣才有代表性。4pC28資訊網——每日最新資訊28at.com

4pC28資訊網——每日最新資訊28at.com

指標

為你的任務選擇正確的指標至關重要。我最喜歡的錯誤使用指標的例子是 Kaggle 的“深空系外行星狩獵”數據集,在那里你可以找到很多筆記本,人們在大約有 5000 個負樣本和 50 個正樣本的嚴重不平衡的數據集上使用準確率。當然,他們得到了 99% 的準確率,并且總是預測負樣本。那樣的話,他們永遠也找不到系外行星,所以讓我們明智地選擇指標。4pC28資訊網——每日最新資訊28at.com

深入討論指標超出了本文的范圍,但我將簡要提及一些可靠的選項:4pC28資訊網——每日最新資訊28at.com

  • F1 分數
  • 精確度和召回率
  • mAP(檢測任務)
  • IoU(分割任務)
  • 準確率(對于平衡的數據集)
  • ROC-AUC

真實圖像分類問題的分數示例:4pC28資訊網——每日最新資訊28at.com

+--------+----------+--------+-----------+--------+| split  | accuracy |   f1   | precision | recall |+--------+----------+--------+-----------+--------+| val    | 0.9915   | 0.9897 | 0.9895    | 0.99   || test   | 0.9926   | 0.9921 | 0.9927    | 0.9915 |+--------+----------+--------+-----------+--------+

4pC28資訊網——每日最新資訊28at.com

為你的任務選擇幾個指標:4pC28資訊網——每日最新資訊28at.com

def get_metrics(gt_labels: List[int], preds: List[int]) -> Dict[str, float]:    num_classes = len(set(gt_labels))    if num_classes == 2:        average = "binary"    else:        average = "macro"    metrics = {}    metrics["accuracy"] = accuracy_score(gt_labels, preds)    metrics["f1"] = f1_score(gt_labels, preds, average=average)    metrics["precision"] = precision_score(gt_labels, preds, average=average)    metrics["recall"] = recall_score(gt_labels, preds, average=average)    return metrics

4pC28資訊網——每日最新資訊28at.com

此外,繪制精確度-閾值和召回率-閾值曲線,以更好地選擇置信度閾值。4pC28資訊網——每日最新資訊28at.com

http://www.www897cc.com/showinfo-26-112733-0.htmlPyTorch 訓練,除了會訓練還要了解這些

聲明:本網頁內容旨在傳播知識,若有侵權等問題請及時與本網聯系,我們將在第一時間刪除處理。郵件:2376512515@qq.com

上一篇: Asp.Net Core實戰-JWT詳解

下一篇: Python 輸入輸出知識點及用法講解

標簽:
  • 熱門焦點
Top 主站蜘蛛池模板: 台安县| 大渡口区| 石门县| 响水县| 白朗县| 金华市| 微山县| 虞城县| 海阳市| 洪洞县| 嘉祥县| 靖西县| 民勤县| 双城市| 南和县| 汪清县| 澄城县| 蒲城县| 慈溪市| 张掖市| 闸北区| 伊春市| 麟游县| 达孜县| 闸北区| 宝清县| 随州市| 长治市| 江安县| 响水县| 龙岩市| 沈阳市| 吴江市| 淅川县| 永顺县| 临武县| 墨江| 浦江县| 酒泉市| 隆林| 乌拉特后旗|