Skip to content
This repository was archived by the owner on Apr 29, 2024. It is now read-only.

Commit 53b3b4a

Browse files
authored
Added metric groups (#25)
1 parent 2a02e0c commit 53b3b4a

File tree

1 file changed

+16
-0
lines changed
  • kilroy_module_pytorch_py_sdk/src/kilroy_module_pytorch_py_sdk/modules

1 file changed

+16
-0
lines changed

kilroy_module_pytorch_py_sdk/src/kilroy_module_pytorch_py_sdk/modules/reward.py

+16
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,10 @@ def name(cls) -> str:
6262
def label(cls) -> str:
6363
return "Supervised Loss"
6464

65+
@classproperty
66+
def group(cls) -> str:
67+
return "1"
68+
6569
@classproperty
6670
def config(cls) -> Dict[str, Any]:
6771
return {
@@ -86,6 +90,10 @@ def name(cls) -> str:
8690
def label(cls) -> str:
8791
return "Reinforced Score"
8892

93+
@classproperty
94+
def group(cls) -> str:
95+
return "1"
96+
8997
@classproperty
9098
def config(cls) -> Dict[str, Any]:
9199
return {
@@ -110,6 +118,10 @@ def name(cls) -> str:
110118
def label(cls) -> str:
111119
return "Reward Model Loss"
112120

121+
@classproperty
122+
def group(cls) -> str:
123+
return "1"
124+
113125
@classproperty
114126
def config(cls) -> Dict[str, Any]:
115127
return {
@@ -134,6 +146,10 @@ def name(cls) -> str:
134146
def label(cls) -> str:
135147
return "Reward Model Score"
136148

149+
@classproperty
150+
def group(cls) -> str:
151+
return "2"
152+
137153
@classproperty
138154
def config(cls) -> Dict[str, Any]:
139155
return {

0 commit comments

Comments
 (0)