Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
}
content[Estimator.SGDRegressor.value] = {
"import": "from sklearn.linear_model import SGDRegressor",
"doc_link": "https://scikit-learn.org\
/stable/modules/generated/sklearn.linear\
_model.SGDRegressor.html#sklearn.linear_model.SGDRegressor",
"evaluation": "regression",
}
content[Estimator.SVR.value] = {
"import": "from sklearn.svm import SVR",
"doc_link": "https://scikit-learn.org/\
stable/modules/generated/sklearn.svm.\
SVR.html#sklearn.svm.SVR",
"evaluation": "regression",
}
content[Estimator.MiniBatchKMeans.value] = {
"import": "from sklearn.cluster import MiniBatchKMeans",
"doc_link": "https://scikit-learn.org/\
stable/modules/clustering.html#mini-batch-k-means",
"evaluation": "classification",
}
content[Estimator.KMeans.value] = {
"import": "from sklearn.cluster import KMeans",
"doc_link": "https://scikit-learn.org/\
stable/modules/clustering.html#k-means",
"evaluation": "clustering",
}
metrics = {
"regression": [
{