Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Select an option

  • Save robintux/841abf0dd322690f220a34fb4ac265ff to your computer and use it in GitHub Desktop.

Select an option

Save robintux/841abf0dd322690f220a34fb4ac265ff to your computer and use it in GitHub Desktop.
# modulos
import os
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

# sklearn
from sklearn.model_selection import train_test_split
from sklearn import metrics

# Redes Neuronales: Keras
from  keras.models import Sequential
from keras.layers import Dense

# Descargamos el dataset de kaggle
!kaggle datasets download robintux/salud-mental-agotamiento-estudiantil

# Descomprimirlo
!unzip salud-mental-agotamiento-estudiantil.zip

# Cargar en memoria
df = pd.read_csv("Salud_mental_agotamiento_estudiantil.csv")
df = df.drop("Unnamed: 0", axis = 1)

df = df[df.dropout_risk >= 0.1].reset_index(drop = True)
df = df.sample(frac= 0.1).reset_index(drop = True)

X = df.iloc[:, :-1]
y = df.dropout_risk
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment