refactor: reorganize project structure and enhance model verbosity
This commit is contained in:
@@ -25,6 +25,7 @@ class SVMModel(TraditionalModel):
|
||||
gamma=params.get("gamma", "scale"),
|
||||
probability=True, # Enable probability prediction
|
||||
random_state=self.config.random_seed,
|
||||
verbose=2
|
||||
)
|
||||
|
||||
return Pipeline([("vectorizer", vectorizer), ("classifier", classifier)])
|
||||
|
||||
Reference in New Issue
Block a user