diff --git a/README.md b/README.md index b863b5f..f0b9fab 100644 --- a/README.md +++ b/README.md @@ -11,3 +11,12 @@ Such as: ``` python main.py --stock_symbol=SP500 --train --input_size=5 --lstm_size=128 --max_epoch=50 ``` + + +Install Conda https://conda.io/docs/installation.html . +Cheatsheet https://conda.io/docs/_downloads/conda-cheatsheet.pdf . + . +## Python 2.7 +conda create -n tensorflow-p2 python=2.7 +source activate tensorflow-p2 +conda install numpy pandas matplotlib tensorflow jupyter notebook scipy scikit-learn nb_conda nltk diff --git a/main.py b/main.py index 8d91447..4676051 100644 --- a/main.py +++ b/main.py @@ -57,7 +57,7 @@ def load_sp500(input_size, num_steps, k=None, target_symbol=None, test_ratio=0.0 print info['file_exists'].value_counts().to_dict() info = info[info['file_exists'] == True].reset_index(drop=True) - info = info.sort('market_cap', ascending=False).reset_index(drop=True) + info = info.sort_values('market_cap', ascending=False).reset_index(drop=True) if k is not None: info = info.head(k)