Skip to content

Commit 6f463de

Browse files
committed
remove label number for GO_UniProt classes
1 parent 62a3f45 commit 6f463de

File tree

1 file changed

+0
-27
lines changed

1 file changed

+0
-27
lines changed

chebai/preprocessing/datasets/go_uniprot.py

Lines changed: 0 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -594,11 +594,6 @@ class _GOUniProtOverX(_GOUniProtDataExtractor, ABC):
594594
READER: dr.ProteinDataReader = dr.ProteinDataReader
595595
THRESHOLD: int = None
596596

597-
@property
598-
@abstractmethod
599-
def label_number(self) -> int:
600-
raise NotImplementedError
601-
602597
@property
603598
def _name(self) -> str:
604599
"""
@@ -706,17 +701,6 @@ class GOUniProtOver250(_GOUniProtOverX):
706701

707702
THRESHOLD: int = 250
708703

709-
def label_number(self) -> int:
710-
"""
711-
Returns the number of labels in the dataset for this threshold.
712-
713-
Overrides the base class method to provide the correct number of labels for a threshold of 100.
714-
715-
Returns:
716-
int: The number of labels (854).
717-
"""
718-
return 854
719-
720704

721705
class GOUniProtOver50(_GOUniProtOverX):
722706
"""
@@ -729,14 +713,3 @@ class GOUniProtOver50(_GOUniProtOverX):
729713
"""
730714

731715
THRESHOLD: int = 50
732-
733-
def label_number(self) -> int:
734-
"""
735-
Returns the number of labels in the dataset for this threshold.
736-
737-
Overrides the base class method to provide the correct number of labels for a threshold of 50.
738-
739-
Returns:
740-
int: The number of labels (1332).
741-
"""
742-
return 1332

0 commit comments

Comments
 (0)