tidying script to run from cmd and via ssh

This commit is contained in:
Tanushree Tunstall 2022-05-28 09:40:24 +01:00
parent 0a84a4b4dc
commit b6f0308e42
4 changed files with 271 additions and 76 deletions

View file

@ -117,15 +117,15 @@ print(len(X_enn)) #53
#------------------------------
# Determine categorical and numerical features
numerical_ix = input_df.select_dtypes(include=['int64', 'float64']).columns
numerical_ix = X.select_dtypes(include=['int64', 'float64']).columns
numerical_ix
num_featuresL = list(numerical_ix)
numerical_colind = input_df.columns.get_indexer(list(numerical_ix) )
numerical_colind = X.columns.get_indexer(list(numerical_ix) )
numerical_colind
categorical_ix = input_df.select_dtypes(include=['object', 'bool']).columns
categorical_ix = X.select_dtypes(include=['object', 'bool']).columns
categorical_ix
categorical_colind = input_df.columns.get_indexer(list(categorical_ix))
categorical_colind = X.columns.get_indexer(list(categorical_ix))
categorical_colind
k_sm = 5 # 5 is deafult