Commit 2ab82592 by Brian McMahan

small fixes; seed settings and readme instructions

parent 3b2e470a
......@@ -25,4 +25,4 @@ If running outside of docker, can do the following from root dir (where you can
jupyter notebook --notebook-dir=$(pwd)
```
Assuming all necessary things are installed. Required Python packages are in `requirements.txt`, except for PyTorch. Follow the installation instructions on the PyTorch website. Also visit [dl4nlp.info](http://dl4nlp.info/en/latest/) for more information about the current training session and [nlproc.info](http://nlproc.info/) for more resources from us!
Assuming all necessary things are installed. Required Python packages are in `requirements.txt`, except for PyTorch. Also visit [dl4nlp.info](http://dl4nlp.info/en/latest/) for more information about the current training session and [nlproc.info](http://nlproc.info/) for more resources from us!
......@@ -153,8 +153,8 @@
"metadata": {},
"outputs": [],
"source": [
"def fix_imbalance(df):\n",
" #np.random.seed(seed)\n",
"def fix_imbalance(df, seed=0):\n",
" np.random.seed(seed)\n",
" median = df.groupby('nationality').agg('count').split.median()\n",
" for nationality in df.nationality.unique():\n",
" \n",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment