Revert "新建仓库维护数据预测项目"

This reverts commit 516126d2a5.
This commit is contained in:
2025-10-17 15:08:09 +08:00
parent 516126d2a5
commit c07dbb586d
72 changed files with 0 additions and 82332 deletions

View File

@@ -1,12 +0,0 @@
import torch
import pandas as pd
import json
def tensor_to_excel(tensor,output_path,file_name = "output"):
df = pd.DataFrame(tensor.numpy())
df.to_excel(output_path + '/' + file_name + '.xlsx', index=False)
def tensor_to_json(tensor,output_path,file_name = "output"):
tensor_list = tensor.tolist() # 转换为Python列表
with open(output_path + '/' + file_name +'.json', 'w') as f:
json.dump(tensor_list, f) # 序列化为JSON

View File

@@ -1,22 +0,0 @@
#1 pytorch 自有格式导出
import torch
def export_model_pt(model,target,name = "model"):
script_model = torch.jit.script(model) # 或 torch.jit.trace(model, input)
script_model.save(target + name + ".pt")
#2 通用格式导出
def export_model_onnx(model,input_tensor,target,name="model"):
torch.onnx.export(model, input_tensor, target+ name + ".onnx")
#3二进制权重导出
def export_model_bin(model,target,name = "weights"):
torch.save(model.state_dict(), target + name + ".bin")
def export_model(model,target,file_name,name):
if name == 'bin':
return export_model_bin(model,target,file_name)
if name == 'onnx':
return export_model_onnx(model,target,file_name)
if name == 'pt':
return export_model_bin(model,target,file_name)
else:
raise ValueError(f"不支持的导出类型")

View File

@@ -1 +0,0 @@
import FC_ML_Data