本文整理汇总了Python中Messages.tune_knn_message方法的典型用法代码示例。如果您正苦于以下问题:Python Messages.tune_knn_message方法的具体用法?Python Messages.tune_knn_message怎么用?Python Messages.tune_knn_message使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类Messages
的用法示例。
在下文中一共展示了Messages.tune_knn_message方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: tune_knn
# 需要导入模块: import Messages [as 别名]
# 或者: from Messages import tune_knn_message [as 别名]
def tune_knn(self, X, y):
msg.tune_knn_message()
k_value = None
msg.loading_message()
k_params = self.mysql_cn.read('select * from params_knn;')
k = k_params['k_value'].tolist()
if not k:
msg.tuning_message()
range_k = list(range(1, 31))
param_grid = {
'n_neighbors': range_k
}
CV_knn = GridSearchCV(estimator=KNN(), param_grid=param_grid, cv=10)
CV_knn.fit(X, y)
k_value_param = CV_knn.best_params_
k_value = k_value_param['n_neighbors']
k = k_value
msg.print_knn_params(k)
msg.insert_message()
self.mysql_cn.insert_update("INSERT INTO params_knn(k_value) VALUES(%d)" % k_value)
return k
elif self.check_tune[0]:
msg.tuning_message()
range_k = list(range(1, 31))
param_grid = {
'n_neighbors': range_k
}
CV_knn = GridSearchCV(estimator=KNN(), param_grid=param_grid, cv=10)
CV_knn.fit(X, y)
k_value_param = CV_knn.best_params_
k_value = k_value_param['n_neighbors']
k = k_value
msg.print_knn_params(k)
msg.update_message()
self.mysql_cn.insert_update("UPDATE params_knn SET k_value=%d" % k_value)
return k
else:
msg.loading_message()
new_k = self.mysql_cn.read('select * from params_knn;')
k_value = new_k['k_value'].tolist()
k = k_value[0]
msg.print_knn_params(k)
return k