Uni/machine-learning
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

run 10K ca.8h with 4 jobs

Browse Source
This commit is contained in:
Maximilian Kany
2025-08-20 20:10:58 +02:00
parent 10dea1cf26
commit d1049e233e
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
compare_models_10k.py
View File

@@ -29,7 +29,7 @@ from sklearn.neural_network import MLPClassifier
set_config(transform_output="pandas") # dataframe supremacy
jobs = 12
jobs = 4
max_iter = 3000
min_entries = 5
@@ -121,14 +121,14 @@ datasets = [
estimators = {
#"RidgeClassifier": RidgeClassifier(random_state=0, max_iter=max_iter),
#"PassiveAggressiveClassifier": PassiveAggressiveClassifier(random_state=0, max_iter=max_iter),
"Perceptron": Perceptron(random_state=0, max_iter=max_iter),
"SGDClassifier": SGDClassifier(random_state=0, max_iter=max_iter),
"NearestCentroid": NearestCentroid(),
"LinearSVC": LinearSVC(random_state=0, max_iter=max_iter),
"GradientBoostingClassifier": GradientBoostingClassifier(random_state=0),
#"Perceptron": Perceptron(random_state=0, max_iter=max_iter),
#"SGDClassifier": SGDClassifier(random_state=0, max_iter=max_iter),
#"NearestCentroid": NearestCentroid(),
#"LinearSVC": LinearSVC(random_state=0, max_iter=max_iter),
#"GradientBoostingClassifier": GradientBoostingClassifier(random_state=0),
"HistGradientBoostingClassifier": HistGradientBoostingClassifier(random_state=0, max_iter=max_iter),
"LinearDiscriminantAnalysis": LinearDiscriminantAnalysis(),
"MLPClassifier": MLPClassifier(random_state=0, max_iter=int(max_iter/20), early_stopping=True),
#"LinearDiscriminantAnalysis": LinearDiscriminantAnalysis(),
#"MLPClassifier": MLPClassifier(random_state=0, max_iter=int(max_iter/20), early_stopping=True),
}
#"VotingClassifier": VotingClassifier(estimators=[('lr', LogisticRegression()), ('rf', RandomForestClassifier())]),