__init__.py 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. # Copyright The Lightning team.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. from torchmetrics.regression.concordance import ConcordanceCorrCoef
  15. from torchmetrics.regression.cosine_similarity import CosineSimilarity
  16. from torchmetrics.regression.crps import ContinuousRankedProbabilityScore
  17. from torchmetrics.regression.csi import CriticalSuccessIndex
  18. from torchmetrics.regression.explained_variance import ExplainedVariance
  19. from torchmetrics.regression.js_divergence import JensenShannonDivergence
  20. from torchmetrics.regression.kendall import KendallRankCorrCoef
  21. from torchmetrics.regression.kl_divergence import KLDivergence
  22. from torchmetrics.regression.log_cosh import LogCoshError
  23. from torchmetrics.regression.log_mse import MeanSquaredLogError
  24. from torchmetrics.regression.mae import MeanAbsoluteError
  25. from torchmetrics.regression.mape import MeanAbsolutePercentageError
  26. from torchmetrics.regression.minkowski import MinkowskiDistance
  27. from torchmetrics.regression.mse import MeanSquaredError
  28. from torchmetrics.regression.nrmse import NormalizedRootMeanSquaredError
  29. from torchmetrics.regression.pearson import PearsonCorrCoef
  30. from torchmetrics.regression.r2 import R2Score
  31. from torchmetrics.regression.rse import RelativeSquaredError
  32. from torchmetrics.regression.spearman import SpearmanCorrCoef
  33. from torchmetrics.regression.symmetric_mape import SymmetricMeanAbsolutePercentageError
  34. from torchmetrics.regression.tweedie_deviance import TweedieDevianceScore
  35. from torchmetrics.regression.wmape import WeightedMeanAbsolutePercentageError
  36. __all__ = [
  37. "ConcordanceCorrCoef",
  38. "ContinuousRankedProbabilityScore",
  39. "CosineSimilarity",
  40. "CriticalSuccessIndex",
  41. "ExplainedVariance",
  42. "JensenShannonDivergence",
  43. "KLDivergence",
  44. "KendallRankCorrCoef",
  45. "LogCoshError",
  46. "MeanAbsoluteError",
  47. "MeanAbsolutePercentageError",
  48. "MeanSquaredError",
  49. "MeanSquaredLogError",
  50. "MinkowskiDistance",
  51. "NormalizedRootMeanSquaredError",
  52. "PearsonCorrCoef",
  53. "R2Score",
  54. "RelativeSquaredError",
  55. "SpearmanCorrCoef",
  56. "SymmetricMeanAbsolutePercentageError",
  57. "TweedieDevianceScore",
  58. "WeightedMeanAbsolutePercentageError",
  59. ]