Skip to content

Commit

Permalink
fix alphabetical order
Browse files Browse the repository at this point in the history
  • Loading branch information
StrikerRUS committed Feb 27, 2020
1 parent 6f02dd8 commit 46e0b8c
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 10 deletions.
6 changes: 3 additions & 3 deletions m2cgen/assemblers/__init__.py
Expand Up @@ -42,7 +42,7 @@
"sklearn.svm.classes.SVC": SVMModelAssembler,
"sklearn.svm.classes.SVR": SVMModelAssembler,

# lightning SVM
# Lightning SVM
"lightning.impl.dual_cd.LinearSVC": SklearnLinearModelAssembler,
"lightning.impl.dual_cd.LinearSVR": SklearnLinearModelAssembler,

Expand Down Expand Up @@ -88,7 +88,7 @@
"statsmodels.base.elastic_net.RegularizedResultsWrapper":
StatsmodelsLinearModelAssembler,

# lightning Linear Regressors
# Lightning Linear Regressors
"lightning.impl.adagrad.AdaGradRegressor": SklearnLinearModelAssembler,
"lightning.impl.primal_cd.CDRegressor": SklearnLinearModelAssembler,
"lightning.impl.fista.FistaRegressor": SklearnLinearModelAssembler,
Expand All @@ -112,7 +112,7 @@
"sklearn.linear_model.stochastic_gradient.SGDClassifier":
SklearnLinearModelAssembler,

# lightning Linear Classifiers
# Lightning Linear Classifiers
"lightning.impl.adagrad.AdaGradClassifier": SklearnLinearModelAssembler,
"lightning.impl.primal_cd.CDClassifier": SklearnLinearModelAssembler,
"lightning.impl.fista.FistaClassifier": SklearnLinearModelAssembler,
Expand Down
14 changes: 7 additions & 7 deletions tests/e2e/test_e2e.py
Expand Up @@ -191,7 +191,7 @@ def classification_binary_random(model):
classification(svm.LinearSVC(random_state=RANDOM_SEED)),
classification_binary(svm.LinearSVC(random_state=RANDOM_SEED)),
# lightning Linear SVM
# Lightning Linear SVM
regression(light_reg.LinearSVR(random_state=RANDOM_SEED)),
classification(light_clf.LinearSVC(
criterion="accuracy", random_state=RANDOM_SEED)),
Expand Down Expand Up @@ -274,7 +274,7 @@ def classification_binary_random(model):
len(utils.get_regression_model_trainer().y_train))),
fit_regularized=STATSMODELS_LINEAR_REGULARIZED_PARAMS))),
# lightning Linear Regression
# Lightning Linear Regression
regression(light_reg.AdaGradRegressor(random_state=RANDOM_SEED)),
regression(light_reg.CDRegressor(random_state=RANDOM_SEED)),
regression(light_reg.FistaRegressor()),
Expand Down Expand Up @@ -311,7 +311,7 @@ def classification_binary_random(model):
classification_binary(linear_model.SGDClassifier(
random_state=RANDOM_SEED)),
# lightning Linear Classifiers
# Lightning Linear Classifiers
classification(light_clf.AdaGradClassifier(
random_state=RANDOM_SEED)),
classification(light_clf.CDClassifier(
Expand All @@ -320,11 +320,11 @@ def classification_binary_random(model):
penalty="l1/l2", multiclass=True, random_state=RANDOM_SEED)),
classification(light_clf.FistaClassifier()),
classification(light_clf.FistaClassifier(multiclass=True)),
classification(light_clf.SDCAClassifier(
classification(light_clf.SAGAClassifier(
random_state=RANDOM_SEED)),
classification(light_clf.SAGClassifier(
random_state=RANDOM_SEED)),
classification(light_clf.SAGAClassifier(
classification(light_clf.SDCAClassifier(
random_state=RANDOM_SEED)),
classification(light_clf.SGDClassifier(
random_state=RANDOM_SEED)),
Expand All @@ -336,11 +336,11 @@ def classification_binary_random(model):
classification_binary(light_clf.CDClassifier(
random_state=RANDOM_SEED)),
classification_binary(light_clf.FistaClassifier()),
classification_binary(light_clf.SDCAClassifier(
classification_binary(light_clf.SAGAClassifier(
random_state=RANDOM_SEED)),
classification_binary(light_clf.SAGClassifier(
random_state=RANDOM_SEED)),
classification_binary(light_clf.SAGAClassifier(
classification_binary(light_clf.SDCAClassifier(
random_state=RANDOM_SEED)),
classification_binary(light_clf.SGDClassifier(
random_state=RANDOM_SEED)),
Expand Down

0 comments on commit 46e0b8c

Please sign in to comment.