Transforme uma tabela mista em perguntas “se/então”.

04 · CÓDIGO COMPLETO E EXECUTÁVEL

Treine e use uma árvore com números e textos.

Copie o programa inteiro. Ele recebe dados mistos sem codificação numérica manual, calcula candidatos e Gini, constrói a árvore, avalia o teste e classifica um novo cliente.

// ============================================================
// ÁRVORE DE DECISÃO: [idade, visitas, plano, comprou]
// Colunas 0 e 1 são numéricas; a coluna 2 contém texto.
// ============================================================
var treino = [
  [22, 1, 'Básico', 'Não'], [25, 2, 'Básico', 'Não'],
  [31, 4, 'Premium', 'Sim'], [35, 5, 'Premium', 'Sim'],
  [42, 2, 'Básico', 'Não'], [45, 6, 'Premium', 'Sim'],
  [52, 5, 'Premium', 'Sim'], [55, 1, 'Básico', 'Não']
];
var teste = [
  [28, 1, 'Básico', 'Não'], [38, 5, 'Premium', 'Sim'],
  [48, 2, 'Básico', 'Não'], [50, 6, 'Premium', 'Sim']
];
var profundidadeMaxima = 3;

function classeMajoritaria(linhas) {
  var quantidadeSim = 0;
  for (var i = 0; i < linhas.length; i++) {
    if (linhas[i][3] === 'Sim') quantidadeSim++;
  }
  if (quantidadeSim >= linhas.length - quantidadeSim) return 'Sim';
  return 'Não';
}

function calcularGini(linhas) {
  if (linhas.length === 0) return 0;
  var quantidadeSim = 0;
  for (var i = 0; i < linhas.length; i++) {
    if (linhas[i][3] === 'Sim') quantidadeSim++;
  }
  var pSim = quantidadeSim / linhas.length;
  var pNao = 1 - pSim;
  return 1 - pSim * pSim - pNao * pNao;
}

// Candidato = [coluna, tipo, valor].
function criarCandidatos(linhas) {
  var candidatos = [];
  for (var coluna = 0; coluna < 3; coluna++) {
    for (var i = 0; i < linhas.length; i++) {
      var repetido = false;
      for (var j = 0; j < candidatos.length; j++) {
        if (candidatos[j][0] === coluna && candidatos[j][2] === linhas[i][coluna]) repetido = true;
      }
      if (!repetido) {
        var tipo = coluna < 2 ? 'numero' : 'texto';
        candidatos.push([coluna, tipo, linhas[i][coluna]]);
      }
    }
  }
  return candidatos;
}

function separar(linhas, candidato) {
  var esquerda = [];
  var direita = [];
  for (var i = 0; i < linhas.length; i++) {
    var atende;
    if (candidato[1] === 'numero') atende = linhas[i][candidato[0]] <= candidato[2];
    else atende = linhas[i][candidato[0]] === candidato[2];
    if (atende) esquerda.push(linhas[i]);
    else direita.push(linhas[i]);
  }
  return [esquerda, direita];
}

// Melhor divisão = [coluna, tipo, valor, esquerda, direita, ganho].
function melhorDivisao(linhas) {
  var candidatos = criarCandidatos(linhas);
  var giniPai = calcularGini(linhas);
  var melhor = null;

  for (var i = 0; i < candidatos.length; i++) {
    var grupos = separar(linhas, candidatos[i]);
    if (grupos[0].length === 0 || grupos[1].length === 0) continue;
    var giniFilhos = (grupos[0].length * calcularGini(grupos[0])
      + grupos[1].length * calcularGini(grupos[1])) / linhas.length;
    var ganho = giniPai - giniFilhos;
    if (melhor === null || ganho > melhor[5]) {
      melhor = [candidatos[i][0], candidatos[i][1], candidatos[i][2],
        grupos[0], grupos[1], ganho];
    }
  }
  return melhor;
}

// Folha = ['folha', classe]. Nó = ['no', coluna, tipo, valor, esquerda, direita].
function construirNo(linhas, profundidade) {
  if (calcularGini(linhas) === 0 || profundidade === profundidadeMaxima) {
    return ['folha', classeMajoritaria(linhas)];
  }
  var melhor = melhorDivisao(linhas);
  if (melhor === null || melhor[5] <= 0) return ['folha', classeMajoritaria(linhas)];
  return ['no', melhor[0], melhor[1], melhor[2],
    construirNo(melhor[3], profundidade + 1),
    construirNo(melhor[4], profundidade + 1)];
}

function classificar(no, entrada) {
  if (no[0] === 'folha') return no[1];
  var atende;
  if (no[2] === 'numero') atende = entrada[no[1]] <= no[3];
  else atende = entrada[no[1]] === no[3];
  if (atende) return classificar(no[4], entrada);
  return classificar(no[5], entrada);
}

function avaliar(arvore, dados) {
  var acertos = 0;
  for (var i = 0; i < dados.length; i++) {
    if (classificar(arvore, dados[i]) === dados[i][3]) acertos++;
  }
  return acertos / dados.length;
}

// ----------------------- USO DO MODELO -----------------------
var arvore = construirNo(treino, 0);
console.log('Árvore:', JSON.stringify(arvore));
console.log('Acurácia de teste:', avaliar(arvore, teste));
console.log('Novo cliente:', classificar(arvore, [40, 5, 'Premium']));