ftnext / atmaCup10-paintings-likes

https://www.guruguru.science/competitions/16/
MIT License
0 stars 0 forks source link

scikit-learnが提供するアルゴリズムを一通り試せる #16

Open ftnext opened 3 years ago

ftnext commented 3 years ago

どのモデルが有望そうかを判断するために、アルゴリズムを変えて訓練を繰り返し回したい

https://www.kaggle.com/startupsci/titanic-data-science-solutions で使っているのと同じくらい試せるようにしたい

# machine learning(パッケージ名のアルファベット順に並び替え)
from sklearn.ensemble import RandomForestClassifier
from sklearn.linear_model import LogisticRegression
from sklearn.linear_model import Perceptron  # classification only
from sklearn.linear_model import SGDClassifier
from sklearn.naive_bayes import GaussianNB  # classification only
from sklearn.neighbors import KNeighborsClassifier
from sklearn.svm import SVC, LinearSVC
from sklearn.tree import DecisionTreeClassifier
ftnext commented 3 years ago

上記classifierに対応するregressor

sklearn.ensemble.RandomForestRegressor https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html

sklearn.linear_model.LinearRegression https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LinearRegression.html

sklearn.linear_model.SGDRegressor https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.SGDRegressor.html

(sklearn.linear_model.ElasticNet) https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.ElasticNet.html

sklearn.neighbors.KNeighborsRegressor https://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KNeighborsRegressor.html

sklearn.svm.SVR https://scikit-learn.org/stable/modules/generated/sklearn.svm.SVR.html

sklearn.svm.LinearSVR https://scikit-learn.org/stable/modules/generated/sklearn.svm.LinearSVR.html

sklearn.tree.DecisionTreeRegressor https://scikit-learn.org/stable/modules/generated/sklearn.tree.DecisionTreeRegressor.html

追加したい sklearn.neural_network.MLPRegressor https://scikit-learn.org/stable/modules/generated/sklearn.neural_network.MLPRegressor.html