Esempio n. 1
0
        public void classificando_email()
        {
            var email = new EmailRecebido("email", "email");
            const int classeEsperada = 1;
            var categoriaEsperada = new Categoria("Esperada");
            geradorDadosTreinamento.ObterCategoria(classeEsperada).Returns(categoriaEsperada);

            var caracteristicas = new[] {2.6, 1.9};
            geradorCaracteristicas.Extrair(email).Returns(caracteristicas);
            algoritmoClassificacao.Classificar(caracteristicas).Returns(classeEsperada);

            var categoria = classificadorEmail.Classificar(email);

            algoritmoClassificacao.Received().Classificar(caracteristicas);
            categoria.Should().Be(categoriaEsperada);
        }
Esempio n. 2
0
        public void treinando_algoritmo_de_classificacao()
        {
            var emailsRecebidos = new[]
            {
                new EmailRecebido("email", "email"),
                new EmailRecebido("email", "email"),
                new EmailRecebido("email", "email")
            };

            var dadosTreinamento = new DadosTreinamento();
            geradorDadosTreinamento.Extrair(emailsRecebidos).Returns(dadosTreinamento);

            classificadorEmail.Treinar(emailsRecebidos);

            geradorDadosTreinamento.Received().Extrair(emailsRecebidos);
            algoritmoClassificacao.Received().Treinar(dadosTreinamento);
        }