parent
2d274e7798
commit
2edd7b08b6
Binary file not shown.
@ -0,0 +1,83 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# encoding: utf-8
|
||||||
|
# @author: 原凯峰
|
||||||
|
# @contact: 2894340009@qq.com
|
||||||
|
# @software: pycharm
|
||||||
|
# @file: MachineLearningDivider.py
|
||||||
|
# @time: 2024/6/26 8:21
|
||||||
|
# @desc:利用随机森林法进行模型训练,能够通过平均响应时间、故障率等数据计算出服务器的健康状态
|
||||||
|
|
||||||
|
|
||||||
|
from sklearn.model_selection import train_test_split
|
||||||
|
from sklearn.ensemble import RandomForestClassifier
|
||||||
|
from sklearn.metrics import classification_report, accuracy_score
|
||||||
|
import pickle
|
||||||
|
def trainmodel():
|
||||||
|
# 假设我们有以下数据集
|
||||||
|
X = [
|
||||||
|
[0.3, 0.005], # 服务器特征:平均响应时间和故障率
|
||||||
|
[2.5, 0.03],
|
||||||
|
[0.7, 0.045],
|
||||||
|
[1.2, 0.002],
|
||||||
|
[3.5, 0.1],
|
||||||
|
[1.3, 0.05],
|
||||||
|
[0.01, 0.15], # 服务器特征:平均响应时间和故障率
|
||||||
|
[5, 0.03],
|
||||||
|
[0.7, 0.015],
|
||||||
|
[1.4, 0.02],
|
||||||
|
[0.15, 0.2],
|
||||||
|
[1.3, 0.005],
|
||||||
|
|
||||||
|
]
|
||||||
|
y = ['良好', '一般', '一般', '良好', '差', '一般', '一般', '差', '良好', '差', '差', '良好'] # 对应的健康状态标签
|
||||||
|
|
||||||
|
# 将健康状态标签转换为数值
|
||||||
|
label_mapping = {'一般': 0, '良好': 1, '差': 2}
|
||||||
|
y_encoded = [label_mapping[label] for label in y]
|
||||||
|
|
||||||
|
# 划分训练集和测试集
|
||||||
|
X_train, X_test, y_train, y_test = train_test_split(X, y_encoded, test_size=0.4, random_state=42)
|
||||||
|
|
||||||
|
# 选择模型,这里使用随机森林分类器
|
||||||
|
model = RandomForestClassifier(n_estimators=100, random_state=42)
|
||||||
|
|
||||||
|
# 训练模型
|
||||||
|
model.fit(X_train, y_train)
|
||||||
|
|
||||||
|
# 预测测试集
|
||||||
|
y_pred = model.predict(X_test)
|
||||||
|
|
||||||
|
# 评估模型
|
||||||
|
print(classification_report(y_test, y_pred))
|
||||||
|
print("Accuracy:", accuracy_score(y_test, y_pred))
|
||||||
|
|
||||||
|
# 保存模型
|
||||||
|
with open('server_health_model.pkl', 'wb') as file:
|
||||||
|
pickle.dump(model, file)
|
||||||
|
|
||||||
|
|
||||||
|
# trainmodel()
|
||||||
|
# 定义一个函数来加载模型并进行预测
|
||||||
|
def load_model_and_predict(new_data):
|
||||||
|
with open('../LogAnalyze/server_health_model.pkl', 'rb') as file:
|
||||||
|
loaded_model = pickle.load(file)
|
||||||
|
predictions = loaded_model.predict(new_data)
|
||||||
|
return predictions
|
||||||
|
|
||||||
|
# 定义一个函数来将预测结果转换为健康等级
|
||||||
|
def predict_health_status(new_data):
|
||||||
|
label_mapping = {'一般': 0, '良好': 1, '差': 2}
|
||||||
|
predictions = load_model_and_predict(new_data)
|
||||||
|
# 创建逆向映射字典
|
||||||
|
inverse_label_mapping = {value: key for key, value in label_mapping.items()}
|
||||||
|
# 使用逆向映射字典转换预测结果
|
||||||
|
health_status = [inverse_label_mapping[pred] for pred in predictions]
|
||||||
|
return health_status
|
||||||
|
|
||||||
|
# 测试函数
|
||||||
|
def testcase():
|
||||||
|
new_data = [[0.4, 0.01]] # 新的服务器数据
|
||||||
|
health_status = predict_health_status(new_data)
|
||||||
|
print("预测的健康状态:", health_status)
|
||||||
|
|
||||||
|
# testcase()
|
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@ -0,0 +1,22 @@
|
|||||||
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
|
# 定义时间格式
|
||||||
|
date_format = "%Y-%m-%d"
|
||||||
|
|
||||||
|
# 获取当前日期,并按照定义的格式转换
|
||||||
|
current_date = datetime.now().strftime(date_format)
|
||||||
|
# 去除前导零
|
||||||
|
current_date = current_date.replace("-0", "-")
|
||||||
|
# 打印当前日期
|
||||||
|
print("当前日期:", current_date)
|
||||||
|
|
||||||
|
for day in range(1,16):
|
||||||
|
|
||||||
|
# 计算当前日期之前15天的日期
|
||||||
|
delta = timedelta(days=-day)
|
||||||
|
previous_date = (datetime.now() + delta).strftime(date_format)
|
||||||
|
# 去除前导零
|
||||||
|
previous_date = previous_date.replace("-0", "-")
|
||||||
|
print(previous_date)
|
||||||
|
# 打印之前15天的日期
|
||||||
|
print("之前15天的日期:", previous_date)
|
@ -1,31 +0,0 @@
|
|||||||
#include <iostream>
|
|
||||||
#include <mysql/mysql.h>
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
MYSQL *conn;
|
|
||||||
conn = mysql_init(NULL);
|
|
||||||
|
|
||||||
if (conn == NULL) {
|
|
||||||
std::cerr << "MySQL init failed" << std::endl;
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
const char *server = "localhost";
|
|
||||||
const char *user = "rtsw";
|
|
||||||
const char *password = "123456";
|
|
||||||
const char *database = "nginxdb";
|
|
||||||
unsigned int port = 3306; // 使用云数据库提供的端口
|
|
||||||
|
|
||||||
if (mysql_real_connect(conn, server, user, password, database, port, NULL, 0) == NULL) {
|
|
||||||
std::cerr << "Connection error: " << mysql_error(conn) << std::endl;
|
|
||||||
mysql_close(conn);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << "Successfully connected to the database" << std::endl;
|
|
||||||
|
|
||||||
// ... 执行数据库操作 ...
|
|
||||||
|
|
||||||
mysql_close(conn);
|
|
||||||
return 0;
|
|
||||||
}
|
|
@ -1,86 +0,0 @@
|
|||||||
#include <iostream>
|
|
||||||
#include <thread>
|
|
||||||
#include <cstring>
|
|
||||||
#include <sys/socket.h>
|
|
||||||
#include <netinet/in.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
|
|
||||||
#define PORT 8080
|
|
||||||
#define MAX_CLIENTS 5
|
|
||||||
#define BUFFER_SIZE 1024
|
|
||||||
|
|
||||||
// 声明 handle_client 函数
|
|
||||||
void handle_client(int client_socket);
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
int server_fd, new_socket;
|
|
||||||
struct sockaddr_in server_addr, client_addr;
|
|
||||||
socklen_t client_len = sizeof(client_addr);
|
|
||||||
char buffer[BUFFER_SIZE];
|
|
||||||
int opt = 1;
|
|
||||||
|
|
||||||
// 创建套接字
|
|
||||||
server_fd = socket(AF_INET, SOCK_STREAM, 0);
|
|
||||||
if (server_fd < 0) {
|
|
||||||
perror("socket creation failed");
|
|
||||||
exit(EXIT_FAILURE);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 设置选项,允许重用地址
|
|
||||||
if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt))) {
|
|
||||||
perror("setsockopt failed");
|
|
||||||
exit(EXIT_FAILURE);
|
|
||||||
}
|
|
||||||
|
|
||||||
server_addr.sin_family = AF_INET;
|
|
||||||
server_addr.sin_addr.s_addr = INADDR_ANY;
|
|
||||||
server_addr.sin_port = htons(PORT);
|
|
||||||
|
|
||||||
// 绑定
|
|
||||||
if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
|
|
||||||
perror("bind failed");
|
|
||||||
exit(EXIT_FAILURE);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 监听
|
|
||||||
if (listen(server_fd, MAX_CLIENTS) < 0) {
|
|
||||||
perror("listen failed");
|
|
||||||
exit(EXIT_FAILURE);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << "Server listening on port " << PORT << std::endl;
|
|
||||||
|
|
||||||
while (true) {
|
|
||||||
client_len = sizeof(client_addr);
|
|
||||||
new_socket = accept(server_fd, (struct sockaddr *)&client_addr, &client_len);
|
|
||||||
if (new_socket < 0) {
|
|
||||||
perror("accept failed");
|
|
||||||
exit(EXIT_FAILURE);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << "New client connected from " << inet_ntoa(client_addr.sin_addr) << " port " << ntohs(client_addr.sin_port) << std::endl;
|
|
||||||
|
|
||||||
// 创建并启动线程处理客户端
|
|
||||||
std::thread(handle_client, new_socket).detach();
|
|
||||||
}
|
|
||||||
|
|
||||||
close(server_fd);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 定义 handle_client 函数
|
|
||||||
void handle_client(int client_socket) {
|
|
||||||
while (true) {
|
|
||||||
char buffer[BUFFER_SIZE];
|
|
||||||
memset(buffer, 0, BUFFER_SIZE); // 清空缓冲区
|
|
||||||
int nbytes = read(client_socket, buffer, BUFFER_SIZE);
|
|
||||||
if (nbytes <= 0) {
|
|
||||||
std::cout << "Client disconnected." << std::endl;
|
|
||||||
close(client_socket);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
std::cout << "Received message from client: " << buffer << std::endl;
|
|
||||||
// 这里可以添加更多的处理逻辑
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,66 +0,0 @@
|
|||||||
#include <iostream>
|
|
||||||
#include <mysql/mysql.h>
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
MYSQL *conn;
|
|
||||||
conn = mysql_init(NULL);
|
|
||||||
|
|
||||||
if (conn == NULL) {
|
|
||||||
std::cerr << "MySQL init failed" << std::endl;
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
const char *server = "localhost";
|
|
||||||
const char *user = "rtsw";
|
|
||||||
const char *password = "123456";
|
|
||||||
const char *database = "nginxdb";
|
|
||||||
unsigned int port = 3306;
|
|
||||||
|
|
||||||
if (mysql_real_connect(conn, server, user, password, database, port, NULL, 0) == NULL) {
|
|
||||||
std::cerr << "Connection error: " << mysql_error(conn) << std::endl;
|
|
||||||
mysql_close(conn);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << "Successfully connected to the database" << std::endl;
|
|
||||||
|
|
||||||
// 插入数据
|
|
||||||
const char *insert_query = "INSERT INTO users (username, email) VALUES ('newuser', 'newuser@example.com')";
|
|
||||||
if (mysql_query(conn, insert_query)) {
|
|
||||||
std::cerr << "Insert error: " << mysql_error(conn) << std::endl;
|
|
||||||
} else {
|
|
||||||
std::cout << "Insert successful" << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查询数据
|
|
||||||
const char *select_query = "SELECT * FROM users";
|
|
||||||
MYSQL_RES *result = mysql_store_result(conn);
|
|
||||||
if (result) {
|
|
||||||
MYSQL_ROW row;
|
|
||||||
while ((row = mysql_fetch_row(result))) {
|
|
||||||
std::cout << "id: " << row[0] << ", username: " << row[1] << ", email: " << row[2] << std::endl;
|
|
||||||
}
|
|
||||||
mysql_free_result(result);
|
|
||||||
} else {
|
|
||||||
std::cerr << "Select error: " << mysql_error(conn) << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 更新数据
|
|
||||||
const char *update_query = "UPDATE users SET email = 'newuser_updated@example.com' WHERE username = 'newuser'";
|
|
||||||
if (mysql_query(conn, update_query)) {
|
|
||||||
std::cerr << "Update error: " << mysql_error(conn) << std::endl;
|
|
||||||
} else {
|
|
||||||
std::cout << "Update successful" << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
// 删除数据
|
|
||||||
const char *delete_query = "DELETE FROM users WHERE username = 'newuser'";
|
|
||||||
if (mysql_query(conn, delete_query)) {
|
|
||||||
std::cerr << "Delete error: " << mysql_error(conn) << std::endl;
|
|
||||||
} else {
|
|
||||||
std::cout << "Delete successful" << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
mysql_close(conn);
|
|
||||||
return 0;
|
|
||||||
}
|
|
@ -1,22 +0,0 @@
|
|||||||
#ifndef CONNECTOR_CONTROLLER_H
|
|
||||||
#define CONNECTOR_CONTROLLER_H
|
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
#include <thread>
|
|
||||||
#include <vector>
|
|
||||||
#include <map>
|
|
||||||
#include <mutex>
|
|
||||||
#include <atomic>
|
|
||||||
#include <sys/socket.h>
|
|
||||||
#include <netinet/in.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
#include <cstring>
|
|
||||||
|
|
||||||
// 函数声明
|
|
||||||
void StartHttpServer();
|
|
||||||
void ConnectToGateway(const std::string& ip, int port);
|
|
||||||
void KeepConnection(int sock);
|
|
||||||
void HandleHttpRequest(int client_socket);
|
|
||||||
|
|
||||||
#endif // CONNECTOR_CONTROLLER_H
|
|
@ -1,10 +0,0 @@
|
|||||||
#ifndef CONNECTOR_DATABASE_H
|
|
||||||
#define CONNECTOR_DATABASE_H
|
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
#include <mysql/mysql.h>
|
|
||||||
|
|
||||||
// 函数声明
|
|
||||||
void DatabaseOperation();
|
|
||||||
|
|
||||||
#endif // CONNECTOR_DATABASE_H
|
|
@ -1,35 +0,0 @@
|
|||||||
#include <iostream>
|
|
||||||
#include <string>
|
|
||||||
#include "connector_controller.h"
|
|
||||||
#include "connector_database.h"
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
std::string command;
|
|
||||||
bool running = true;
|
|
||||||
|
|
||||||
while (running) {
|
|
||||||
std::cout << "Enter command (start_http_server/connect_gateway/database_operation/exit): ";
|
|
||||||
std::cin >> command;
|
|
||||||
|
|
||||||
if (command == "start_http_server") {
|
|
||||||
StartHttpServer();
|
|
||||||
} else if (command == "connect_gateway") {
|
|
||||||
std::string ip;
|
|
||||||
int port;
|
|
||||||
std::cout << "Enter the gateway IP address: "<<std::endl;
|
|
||||||
std::cin>>ip; // 使用getline以获取包含空格的IP地址
|
|
||||||
std::cout << "Enter the gateway port: "<<std::endl;
|
|
||||||
std::cin >> port;
|
|
||||||
ConnectToGateway(ip, port);
|
|
||||||
} else if (command == "database_operation") {
|
|
||||||
DatabaseOperation();
|
|
||||||
} else if (command == "exit") {
|
|
||||||
running = false;
|
|
||||||
} else {
|
|
||||||
std::cout << "Unknown command" << std::endl;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << "Exiting program." << std::endl;
|
|
||||||
return 0;
|
|
||||||
}
|
|
Binary file not shown.
@ -1,125 +0,0 @@
|
|||||||
#include <iostream>
|
|
||||||
#include <thread>
|
|
||||||
#include <vector>
|
|
||||||
#include <map>
|
|
||||||
#include <mutex>
|
|
||||||
#include <atomic>
|
|
||||||
#include <sys/socket.h>
|
|
||||||
#include <netinet/in.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
|
|
||||||
#define BUFFER_SIZE 1024
|
|
||||||
|
|
||||||
std::vector<std::thread> threads; // 存储所有连接线程
|
|
||||||
std::map<int, int> sockets; // 存储套接字
|
|
||||||
std::mutex sockets_mutex; // 用于同步访问 sockets
|
|
||||||
std::atomic<bool> exit_flag(false); // 原子退出标志
|
|
||||||
|
|
||||||
|
|
||||||
// 维护连接的函数
|
|
||||||
void KeepConnection(int sock) {
|
|
||||||
char buffer[BUFFER_SIZE];
|
|
||||||
int nbytes;
|
|
||||||
while (!exit_flag) {
|
|
||||||
nbytes = recv(sock, buffer, BUFFER_SIZE, 0);
|
|
||||||
if (nbytes <= 0) {
|
|
||||||
std::cout << "Gateway disconnected" << std::endl;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
std::cout << "Received from gateway (Socket " << sock << "): " << buffer << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
std::lock_guard<std::mutex> lock(sockets_mutex);
|
|
||||||
sockets.erase(sock); // 从集合中移除套接字
|
|
||||||
}
|
|
||||||
close(sock);
|
|
||||||
}
|
|
||||||
// 连接到网关的函数
|
|
||||||
void ConnectToGateway(const std::string& ip, int port) {
|
|
||||||
int sock = socket(AF_INET, SOCK_STREAM, 0);
|
|
||||||
if (sock < 0) {
|
|
||||||
perror("Could not create socket");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct sockaddr_in server_addr;
|
|
||||||
server_addr.sin_family = AF_INET;
|
|
||||||
server_addr.sin_port = htons(port);
|
|
||||||
if (inet_pton(AF_INET, ip.c_str(), &server_addr.sin_addr) <= 0) {
|
|
||||||
perror("Invalid address");
|
|
||||||
close(sock);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (connect(sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
|
|
||||||
perror("Connection to the server failed");
|
|
||||||
close(sock);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
std::lock_guard<std::mutex> lock(sockets_mutex);
|
|
||||||
sockets[sock] = 1; // 将套接字添加到集合中
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << "Connected to gateway at " << ip << ":" << port << " (Socket " << sock << ")" << std::endl;
|
|
||||||
|
|
||||||
// 在新线程中维护连接
|
|
||||||
threads.emplace_back(KeepConnection, sock);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
std::string command, ip;
|
|
||||||
int port;
|
|
||||||
bool in_connect_mode = false;
|
|
||||||
|
|
||||||
while (true) {
|
|
||||||
std::cout << "Enter command (connect/exitconnect/exit): ";
|
|
||||||
std::cin >> command;
|
|
||||||
|
|
||||||
if (command == "connect") {
|
|
||||||
if (in_connect_mode) {
|
|
||||||
std::cout << "Already in connect mode." << std::endl;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
in_connect_mode = true; // 进入连接模式
|
|
||||||
std::cout << "Enter the gateway IP address: ";
|
|
||||||
std::cin >> ip;
|
|
||||||
std::cout << "Enter the gateway port: ";
|
|
||||||
std::cin >> port;
|
|
||||||
ConnectToGateway(ip, port);
|
|
||||||
} else if (command == "exitconnect") {
|
|
||||||
if (!in_connect_mode) {
|
|
||||||
std::cout << "Not in connect mode." << std::endl;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
in_connect_mode = false; // 退出连接模式
|
|
||||||
std::cout << "Exiting connect mode." << std::endl;
|
|
||||||
} else if (command == "exit") {
|
|
||||||
exit_flag = true; // 设置退出标志
|
|
||||||
std::cout << "Exiting program." << std::endl;
|
|
||||||
// 关闭所有套接字
|
|
||||||
for (auto& sock_pair : sockets) {
|
|
||||||
shutdown(sock_pair.first, SHUT_RDWR); // 关闭套接字的发送和接收
|
|
||||||
close(sock_pair.first);
|
|
||||||
}
|
|
||||||
sockets.clear();
|
|
||||||
// 等待所有线程结束
|
|
||||||
for (auto& thread : threads) {
|
|
||||||
if (thread.joinable()) {
|
|
||||||
thread.join();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
threads.clear();
|
|
||||||
break; // 退出主循环
|
|
||||||
} else {
|
|
||||||
std::cout << "Unknown command" << std::endl;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
@ -1,74 +0,0 @@
|
|||||||
#!/usr/bin/env python
|
|
||||||
# encoding: utf-8
|
|
||||||
# @author: 原凯峰
|
|
||||||
# @contact: 2894340009@qq.com
|
|
||||||
# @software: pycharm
|
|
||||||
# @file: MachineLearningDivider.py
|
|
||||||
# @time: 2024/6/26 8:21
|
|
||||||
# @desc:利用随机森林法进行模型训练,能够通过平均响应时间、故障率等数据计算出服务器的健康状态
|
|
||||||
|
|
||||||
|
|
||||||
from sklearn.model_selection import train_test_split
|
|
||||||
from sklearn.ensemble import RandomForestClassifier
|
|
||||||
from sklearn.metrics import classification_report, accuracy_score
|
|
||||||
import pickle
|
|
||||||
# 假设我们有以下数据集
|
|
||||||
X = [
|
|
||||||
[0.3, 0.005], # 服务器特征:平均响应时间和故障率
|
|
||||||
[2.5, 0.03],
|
|
||||||
[0.7, 0.045],
|
|
||||||
[1.2, 0.002]
|
|
||||||
]
|
|
||||||
y = ['良好', '差', '差', '良好'] # 对应的健康状态标签
|
|
||||||
|
|
||||||
# 将健康状态标签转换为数值
|
|
||||||
label_mapping = {'一般': 0, '良好': 1, '差': 2, '极差': 3}
|
|
||||||
y_encoded = [label_mapping[label] for label in y]
|
|
||||||
|
|
||||||
# 划分训练集和测试集
|
|
||||||
X_train, X_test, y_train, y_test = train_test_split(X, y_encoded, test_size=0.25, random_state=42)
|
|
||||||
|
|
||||||
# 选择模型,这里使用随机森林分类器
|
|
||||||
model = RandomForestClassifier(n_estimators=100, random_state=42)
|
|
||||||
|
|
||||||
# 训练模型
|
|
||||||
model.fit(X_train, y_train)
|
|
||||||
|
|
||||||
# 预测测试集
|
|
||||||
y_pred = model.predict(X_test)
|
|
||||||
|
|
||||||
# 评估模型
|
|
||||||
print(classification_report(y_test, y_pred))
|
|
||||||
print("Accuracy:", accuracy_score(y_test, y_pred))
|
|
||||||
|
|
||||||
with open('server_health_model.pkl', 'wb') as file:
|
|
||||||
pickle.dump(model, file)
|
|
||||||
|
|
||||||
|
|
||||||
# 保存模型
|
|
||||||
with open('server_health_model.pkl', 'wb') as file:
|
|
||||||
pickle.dump(model, file)
|
|
||||||
|
|
||||||
# 定义一个函数来加载模型并进行预测
|
|
||||||
def load_model_and_predict(new_data):
|
|
||||||
with open('server_health_model.pkl', 'rb') as file:
|
|
||||||
loaded_model = pickle.load(file)
|
|
||||||
predictions = loaded_model.predict(new_data)
|
|
||||||
return predictions
|
|
||||||
|
|
||||||
# 定义一个函数来将预测结果转换为健康等级
|
|
||||||
def predict_health_status(new_data):
|
|
||||||
predictions = load_model_and_predict(new_data)
|
|
||||||
# 创建逆向映射字典
|
|
||||||
inverse_label_mapping = {value: key for key, value in label_mapping.items()}
|
|
||||||
# 使用逆向映射字典转换预测结果
|
|
||||||
health_status = [inverse_label_mapping[pred] for pred in predictions]
|
|
||||||
return health_status
|
|
||||||
|
|
||||||
# 测试函数
|
|
||||||
def testcase():
|
|
||||||
new_data = [[0.4, 0.01]] # 新的服务器数据
|
|
||||||
health_status = predict_health_status(new_data)
|
|
||||||
print("预测的健康状态:", health_status)
|
|
||||||
|
|
||||||
testcase()
|
|
Binary file not shown.
Binary file not shown.
@ -1,2 +0,0 @@
|
|||||||
dic = {'a':1,'b':2,'c':3}
|
|
||||||
for i in dic.keys():
|
|
Loading…
Reference in new issue