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