/
pyoadfe
/
hw7-particles
Обзор
Документация
Войти
/
pyoadfe
/
hw7-particles
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
master
predict.py
24 строки
622 B
Matwey V. Kornilov
Update for 2024
23 май 2024, 13:48
23 май 2024, 13:48
358d615
Код
Авторство
О чём код?
#!/usr/bin/env python3 import numpy as np import pandas as pd import onnxruntime as rt def main(): data = pd.read_csv("test.csv.gz", index_col="ID") x = np.asarray(data, dtype=np.float32) sess = rt.InferenceSession("particles.onnx") input_name = sess.get_inputs()[0].name label_name = sess.get_outputs()[0].name print("Inputs: {}".format([x.name for x in sess.get_inputs()])) print("Outputs: {}".format([x.name for x in sess.get_outputs()])) pred = sess.run([label_name], {input_name: x}) np.savetxt("predict.csv", pred[0], delimiter=",") if __name__ == "__main__": main()