Skip to content

Commit

Permalink
Merge pull request #11 from ArnovanHilten/dev
Browse files Browse the repository at this point in the history
Release 1.2.0
  • Loading branch information
ArnovanHilten authored Aug 13, 2020
2 parents 6dbc1e3 + 13ea850 commit aa68dda
Show file tree
Hide file tree
Showing 62 changed files with 26,580 additions and 26,634 deletions.
22 changes: 12 additions & 10 deletions GenNet.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,16 @@
import sys
import os
import warnings

warnings.filterwarnings('ignore')
import argparse

sys.path.insert(1, os.path.dirname(os.getcwd()) + "/utils/")
from utils.Create_plots import plot
from utils.Train_network import train_classification, train_regression
from utils.Convert import convert


def main(args):
if args.mode == 'train':
if args.problem_type == "classification":
Expand All @@ -23,19 +26,20 @@ def main(args):


if __name__ == '__main__':
parser = argparse.ArgumentParser(description = "GenNet: Interpretable neural networks for phenotype prediction.",
parser = argparse.ArgumentParser(description="GenNet: Interpretable neural networks for phenotype prediction.",
epilog="Check the wiki on github.com/arnovanhilten/gennet/ for more info")
subparsers = parser.add_subparsers(help = "GenNet main options", dest="mode")
subparsers = parser.add_subparsers(help="GenNet main options", dest="mode")

parser_convert = subparsers.add_parser("convert", help = "Convert genotype data to hdf5")
parser_convert = subparsers.add_parser("convert", help="Convert genotype data to hdf5")
parser_convert.add_argument("-g", "--genotype", nargs='+', type=str, help="path/paths to genotype data folder")
parser_convert.add_argument('-study_name', type=str, required=True, nargs='+',
help=' Name for saved genotype data, without ext')
parser_convert.add_argument('-variants', type=str, help="Path to file with row numbers of variants to include, if none is "
"given all variants will be used", default=None)
parser_convert.add_argument('-variants', type=str,
help="Path to file with row numbers of variants to include, if none is "
"given all variants will be used", default=None)
parser_convert.add_argument("-o", "--out", type=str, required=True, help="path to save result folder")
parser_convert.add_argument('-ID', action='store_true', default=False,
help='Flag to convert minimac data to genotype per subject files first (default False)')
help='Flag to convert minimac data to genotype per subject files first (default False)')

parser_convert.add_argument('-vcf', action='store_true', default=False, help='Flag for VCF data to convert')
parser_convert.add_argument('-tcm', type=int, default=500000000, help='Modifier for chunk size during TRANSPOSING'
Expand All @@ -55,7 +59,7 @@ def main(args):
parser_train.add_argument(
"-problem_type",
default='classification', type=str,
choices=['classification','regression'],
choices=['classification', 'regression'],
help="Type of problem, choices are: classification or regression"
)
parser_train.add_argument(
Expand Down Expand Up @@ -110,11 +114,9 @@ def main(args):
type=int,
help="Only for layer weight: Number of the to be plotted layer",
metavar="Layer_number:",
default = 0
default=0
)

args = parser.parse_args()

main(args)


2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ cd ~
python3 -m venv env_GenNet
```

This automatically instals the latest Tensorflow version for which GenNet has been tested. If you have an older version of CUDA install the appriopriate tensorflow-gpu by
This automatically installs the latest Tensorflow version for which GenNet has been tested. If you have an older version of CUDA install the appriopriate tensorflow-gpu by
`pip install tensorflow-gpu==1.13.1` (change 1.13.1 to your version).

**Activate the environment**
Expand Down
Loading

0 comments on commit aa68dda

Please sign in to comment.