added reports and params
This commit is contained in:
@@ -11,15 +11,14 @@ from sets.Data import vertical_split
|
||||
|
||||
class WeightFiltration(Strategy):
|
||||
def __init__(self,
|
||||
target_class_index: int,
|
||||
arch,
|
||||
num_classes: int = 20,
|
||||
epochs: int = 6,
|
||||
lr: float = 100.0,
|
||||
gamma: float = 0.01,
|
||||
lambda_1 = 25
|
||||
|
||||
):
|
||||
target_class_index: int,
|
||||
arch,
|
||||
num_classes: int = 20,
|
||||
epochs: int = 6,
|
||||
lr: float = 100.0,
|
||||
gamma: float = 0.01,
|
||||
lambda_1 = 25
|
||||
):
|
||||
super().__init__(target_class_index=target_class_index)
|
||||
self.epochs = epochs
|
||||
self.lr = lr
|
||||
|
||||
Reference in New Issue
Block a user