clarindasusan commited on
Commit
fc8ddd7
·
verified ·
1 Parent(s): 3bb5f30

Update src/train_model.py

Browse files
Files changed (1) hide show
  1. 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 = 200, n_samples: int = None):
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}")