Classificação de Tipos de Personalidade

Conheça o projeto

Este desafio propõe a construção de um modelo de machine learning capaz de prever o tipo de personalidade de um indivíduo — introvertido, extrovertido ou ambivertido — a partir de 29 traços psicológicos.

Recursos

Materiais para você usar como base para o desenvolvimento

Instruções

Estrutura, regras e requisitos do projeto

O seu objetivo é construir um modelo de machine learning para prever o tipo de personalidade (introvertido (Introvert), extrovertido (Extrovert) ou ambivertido (Ambivert)) de um indivíduo com base em 29 de seus traços psicológicos. O desafio é explorar, treinar e otimizar um modelo utilizando o LightGBM, buscando a máxima performance possível.


📦 Dataset

O dataset para este desafio está disponível publicamente no Kaggle e pode ser encontrado no seguinte link: Introvert, Extrovert & Ambivert Classification - Dataset.

Sinopse do Dataset

O conjunto de dados foi gerado sinteticamente e contém 20.000 entradas de dados, com 30 colunas no total. Cada linha representa um indivíduo, e as colunas descrevem diferentes aspectos da sua personalidade, como energia social, preferência por passar tempo sozinho e níveis de extroversão. A variável-alvo é o tipo de personalidade, que é multiclasse e está balanceada para garantir que o modelo não seja viciado em nenhuma das categorias.


Nome e Descrição das Variáveis do Dataset

O dataset é composto por 29 colunas de características (features) e 1 coluna de destino (target):

Nome da VariávelDescrição
social_energyNível de energia que a pessoa sente em ambientes sociais, variando de 0 a 10.
alone_time_preferenceNível de preferência por passar tempo sozinho, de 0 a 10.
talkativenessNível de propensão para conversar, de 0 a 10.
listening_skillsHabilidade para ouvir os outros, de 0 a 10.
assertivenessNível de firmeza e autoconfiança, de 0 a 10.
conflict_avoidanceNível de preferência por evitar confrontos, de 0 a 10.
new_people_comfortNível de conforto ao conhecer novas pessoas, de 0 a 10.
routine_vs_spontaneityPreferência entre rotina e espontaneidade, de 0 a 10.
emotional_expressionNível de facilidade para expressar emoções, de 0 a 10.
energy_in_large_groupsNível de energia em grandes grupos, de 0 a 10.
friend_circle_sizeTamanho do círculo de amigos, de 0 a 10.
party_enjoymentNível de satisfação em festas, de 0 a 10.
deep_vs_small_talkPreferência entre conversas profundas e superficiais, de 0 a 10.
social_media_usageNível de uso de redes sociais, de 0 a 10.
public_speaking_comfortNível de conforto ao falar em público, de 0 a 10.
introversion_scoreNível de introversão, de 0 a 10.
extroversion_scoreNível de extroversão, de 0 a 10.
ambiversion_scoreNível de ambiversão, de 0 a 10.
creativityNível de criatividade, de 0 a 10.
adaptabilityNível de adaptabilidade a novas situações, de 0 a 10.
planning_tendencyTendência a fazer planos, de 0 a 10.
reflection_tendencyTendência a refletir sobre as coisas, de 0 a 10.
adventure_seekingTendência a buscar aventuras, de 0 a 10.
curiosityNível de curiosidade, de 0 a 10.
orderlinessNível de organização e ordem, de 0 a 10.
stress_managementHabilidade de gerenciar o estresse, de 0 a 10.
long_term_planningHabilidade de planejamento a longo prazo, de 0 a 10.
patienceNível de paciência, de 0 a 10.
trust_in_othersNível de confiança nos outros, de 0 a 10.
personality_typeVariável-alvo: O tipo de personalidade (Introvert, Extrovert ou Ambivert).

Observação

O link do dataset pode ser encontrado na seção de recursos, ou nas instruções do desafio.


Entrega

Após concluir o desafio, você deve enviar a URL do seu código no GitHub para a plataforma.

Além disso, que tal fazer um post no LinkedIn compartilhando o seu aprendizado e contando como foi a experiência?

É uma excelente forma de demonstrar seus conhecimentos e atrair novas oportunidades!

Feito com 💜 por Rocketseat 👋

Tarefas

Use este checklist para ajudar a organizar a sua entrega

Resolução

Confira os resultados esperados do projeto

Paywall background

Envie o projeto para ver a resolução

Ao enviar seu projeto, você poderá conferir os resultados esperados

Projetos relacionados