Spaces:
Sleeping
Sleeping
Update src/train_model.py
Browse files- src/train_model.py +1 -1
src/train_model.py
CHANGED
|
@@ -648,7 +648,7 @@ def evaluate_model(model: FuzzyNeuralNetwork, X: torch.Tensor, y: torch.Tensor)
|
|
| 648 |
}
|
| 649 |
|
| 650 |
|
| 651 |
-
def train_disaster_model(disaster_type: str, epochs: int =
|
| 652 |
print(f"\n{'='*60}")
|
| 653 |
print(f" Training FNN for: {disaster_type.upper()}")
|
| 654 |
print(f"{'='*60}")
|
|
|
|
| 648 |
}
|
| 649 |
|
| 650 |
|
| 651 |
+
def train_disaster_model(disaster_type: str, epochs: int = 1000, n_samples: int = None):
|
| 652 |
print(f"\n{'='*60}")
|
| 653 |
print(f" Training FNN for: {disaster_type.upper()}")
|
| 654 |
print(f"{'='*60}")
|