文件名称：torch-metrics:pytorch中模型评估的指标
文件大小：17KB
文件格式：ZIP
更新时间：2024-05-24 02:09:47
machine-learning computer-vision deep-learning metrics pytorch
火炬指标 PyTorch的模型评估指标 火炬指标作为自定义库，以提供Pytorch共同ML评价指标，类似于tf.keras.metrics 。 如，Pytorch没有用于模型评估指标的内置库torch.metrics 。 这类似于的指标库。 用法 pip install --upgrade torch-metrics from torch_metrics import Accuracy ## define metric ## metric = Accuracy ( from_logits = False ) y_pred = torch . tensor ([ 1 , 2 , 3 , 4 ]) y_true = torch . tensor ([ 0 , 2 , 3 , 4 ]) print ( metric ( y_pred , y_true )) ## define metri
【文件预览】：
torch-metrics-main
----.gitignore(97B)
----torch_metrics()
--------regression()
--------utils.py(526B)
--------classification()
--------__init__.py(53B)
----README.md(2KB)
----.github()
--------stale.yml(2KB)
--------workflows()
--------dependabot.yml(125B)
----.pre-commit-config.yaml(662B)
----tests()
--------test_classification_metrics.py(912B)
--------test_regression_metrics.py(2KB)
--------input_data.py(1KB)
----LICENSE(1KB)
----requirements.txt(106B)
----setup.py(380B)
----Makefile(292B)