Show similarity in app

This commit is contained in:
2025-02-09 23:42:31 +01:00
parent 60a053b929
commit 65faf0230b
7 changed files with 109 additions and 11 deletions

View File

@@ -5,7 +5,8 @@
#include "values/Similarity.hpp"
#include "canvas/Canvas.hpp"
enum DrawingStage{
enum DrawingStage
{
setup,
drawTree,
drawBig,
@@ -46,4 +47,7 @@ private:
void setUpManager();
std::array<float, 3> simil;
std::vector<std::array<float, 3>> similTable;
void setUpTable();
};

View File

@@ -129,6 +129,10 @@ void Vapp::draw()
enableAll = false;
stageOfDrawing = DrawingStage::setup;
}
if (ImGui::MenuItem("CalcTable", nullptr, false, enableAll))
{
setUpTable();
}
ImGui::EndMainMenuBar();
}
@@ -173,6 +177,28 @@ void Vapp::draw()
ImGui::LabelText("##sim1", "cosine_similarity: %f", simil[0]);
ImGui::LabelText("##sim2", "hamming_distance: %f", simil[1]);
ImGui::LabelText("##sim3", "cosine_similarity_int: %f", simil[2]);
const ImGuiTableFlags flags = ImGuiTableFlags_NoHostExtendX | ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_Resizable | ImGuiTableFlags_BordersOuter | ImGuiTableFlags_BordersV | ImGuiTableFlags_ContextMenuInBody;
const int columns = 4;
if (ImGui::BeginTable("table1", columns, flags))
{
for (int row = 0; row < similTable.size(); row++)
{
ImGui::TableNextRow();
for (int column = 0; column < columns; column++)
{
ImGui::TableSetColumnIndex(column);
if (column == 0)
{
ImGui::Text("%d", row);
continue;
}
ImGui::Text("%f", similTable[row][column - 1]);
}
}
ImGui::EndTable();
}
ImGui::End();
}
@@ -232,3 +258,53 @@ void Vapp::setUpManager()
sql::finalize(get_gen_stmt);
}
void Vapp::setUpTable()
{
sqlite3_stmt *get_gen_stmt;
sql::prepare_v2(db, get_gen, -1, &get_gen_stmt, NULL);
DnaManager::cleanUp(&manager, ids[selected_id_index]);
similTable.clear();
while (true)
{
sql::bind_int64(get_gen_stmt, 1, manager.id);
sql::bind_int64(get_gen_stmt, 2, manager.generation);
bool new_gen = false;
while (sql::step(get_gen_stmt) != SQL_DONE)
{
int64_t pos = sql::column_int64(get_gen_stmt, 1);
int64_t liked = sql::column_int64(get_gen_stmt, 2);
UiUnit unit = DnaManager::next(&manager);
if ((unit.index != pos))
{
// DOTO: SET ERROR
TraceLog(LOG_ERROR, "LOADING DNA");
sql::finalize(get_gen_stmt);
return;
}
unit.liked = (Liked)liked;
new_gen = DnaManager::like(unit, &manager);
}
if (new_gen)
{
similTable.emplace_back();
int s = similTable.size() - 1;
similTable[s][0] = Similarity::calc_similarity(manager.vector, Similarity::cosine_similarity);
similTable[s][1] = Similarity::calc_similarity(manager.vector, Similarity::hamming_distance);
similTable[s][2] = Similarity::calc_similarity(manager.vector, Similarity::cosine_similarity_int);
DnaManager::newGen(&manager);
}
else
{
break;
}
sql::reset(get_gen_stmt);
}
sql::finalize(get_gen_stmt);
}