Add prediction.hpp
This commit is contained in:
23
prediction.hpp
Normal file
23
prediction.hpp
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <chrono>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
namespace engine
|
||||||
|
{
|
||||||
|
|
||||||
|
struct ClassScore
|
||||||
|
{
|
||||||
|
std::string label;
|
||||||
|
float probability;
|
||||||
|
int class_index;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct Prediction
|
||||||
|
{
|
||||||
|
std::vector< ClassScore > scores;
|
||||||
|
std::chrono::microseconds total_time{};
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace engine
|
||||||
Reference in New Issue
Block a user