def grupos() -> None:
df_grupos = ler_csv(
f'groups.csv',
usar_colunas=['category.shortname', 'rating', 'group_id'])
db = Database()
for id_grupo, categoria, nota in tqdm(df_grupos.values):
db.set_document('grupos', {
'_id': id_grupo,
'categoria': categoria,
'nota': nota
})
def salvar_modelo(dataset: dict) -> str:
db = Database()
modelo_pk = pickle.dumps(dataset['modelo'])
modelo = {
'modelo': modelo_pk,
'grupo': dataset['grupo'],
'metricas': dataset['metricas'],
'0': dataset['0'],
'1': dataset['1'],
}
return db.set_document('modelo', modelo)