preprocesing fix

This commit is contained in:
Maciej Sobkowiak 2021-04-12 00:30:39 +02:00
parent 171fd416dc
commit 2358985132

View File

@ -1,12 +1,9 @@
import sys import sys
import kaggle
import pandas as pd import pandas as pd
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
sc = pd.read_csv('who_suicide_statistics.csv') sc = pd.read_csv('who_suicide_statistics.csv')
sc
train, validate, test = np.split(sc.sample(frac=1, random_state=42), train, validate, test = np.split(sc.sample(frac=1, random_state=42),
[int(.6*len(sc)), int(.8*len(sc))]) [int(.6*len(sc)), int(.8*len(sc))])