web-dev-qa-db-fra.com

Pourquoi est-ce que j'obtiens NameError: le nom 'train_test_split' n'est pas défini?

J'ai suivi ce tutoriel que j'ai trouvé en ligne sur l'analyse de la parole dans Deep Learning, cela me donnait toujours l'erreur de nom. Je suis assez nouveau en python, donc je ne sais pas comment le définir. mais alors train_test_split est une méthode par défaut pour fractionner les données, train_test_split est importé

voici le code:

'' '

import numpy as np
import pandas as pd 
import os
import seaborn as sns
import matplotlib.pyplot as plt
import seaborn as sns
plt.style.use('fivethirtyeight')
from tqdm import tqdm
print(os.listdir("../input"))

from keras import Sequential
from keras import optimizers
from keras.preprocessing.sequence import pad_sequences
from keras.models import Sequential,Model
from keras.layers import LSTM, Dense, Bidirectional, Input,Dropout,BatchNormalization,CuDNNLSTM, GRU, CuDNNGRU, Embedding, GlobalMaxPooling1D, GlobalAveragePooling1D, Flatten
from keras import backend as K
from keras.engine.topology import Layer
from keras import initializers, regularizers, constraints
from sklearn.model_selection import KFold, cross_val_score, train_test_split

train = pd.read_json('C:/Users/User/Downloads/dont-call-me-turkey/train.json')
display(train.shape)

train.head()

train_train, train_val = train_test_split(train, random_state = 42)
xtrain = [k for k in train_train['audio_embedding']]
ytrain = train_train['is_turkey'].values
xval = [k for k in train_val['audio_embedding']]
yval = train_val['is_turkey'].values '''

il a donné une erreur:

NameError                                 Traceback (most recent call last)
<ipython-input-19-1e07851e6519> in <module>
----> 1 train_train, train_val = train_test_split(train, random_state = 42)
      2 xtrain = [k for k in train_train['audio_embedding']]
      3 ytrain = train_train['is_turkey'].values
      4 xval = [k for k in train_val['audio_embedding']]
      5 yval = train_val['is_turkey'].values

NameError: name 'train_test_split' is not defined
pythondeep-learningjupyter-notebooknameerror
1
silvermaze

Vous n'avez probablement pas installé sklearn Pip install sklearn Si vous l'avez déjà fait, essayez: from sklearn.cross_validation import train_test_split

3
Capie