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