Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merge dev_postgresql into the master branch #1115

Merged
merged 90 commits into from
Oct 25, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
90 commits
Select commit Hold shift + click to select a range
264f9e0
Add DeprecationWarning ignore message to the running script of cnn ex…
wannature Apr 9, 2023
468c34a
Merge pull request #1053 from wannature/singa_v32
chrishkchris Apr 10, 2023
8edcb96
Add training scripts for xceptionnet
Apr 27, 2023
374fef4
Merge pull request #1056 from zmeihui/23-4-27
lzjpaul May 2, 2023
0f16a27
mlp model examples on top of postgresql
wannature May 11, 2023
a39f616
Merge pull request #1057 from wannature/singa_v33
lzjpaul May 13, 2023
df937b0
readme for malaria prediction with cnn model
liuchangshiye May 30, 2023
8aca66c
Merge pull request #1060 from liuchangshiye/malaria-cnn
lzjpaul Jun 2, 2023
2f78a24
add shell command file to run the codes
liuchangshiye Jun 8, 2023
0dce2a4
Merge pull request #1064 from liuchangshiye/malaria-cnn
lzjpaul Jun 8, 2023
12b9204
Init the model selection examples, and add README.md
NLGithubWP May 31, 2023
32f018a
Merge pull request #1065 from NLGithubWP/add_model_selection_psql
lzjpaul Jun 10, 2023
f4d820f
Init armnet examples, add README.md
solopku Jun 12, 2023
8261ada
Merge pull request #1066 from solopku/dev-postgresql
lzjpaul Jun 12, 2023
10a7fc9
Add data loader for malaria dataset
liuchangshiye Jun 13, 2023
1970158
Merge pull request #1070 from liuchangshiye/dev-postgresql
lzjpaul Jul 17, 2023
4257283
add cnn model for malaria detection
lemonviv Jul 20, 2023
c847d46
Merge pull request #1072 from lemonviv/dev-postgresql
lzjpaul Jul 20, 2023
a76f3d7
add mlp model for malaria detection
liuchangshiye Aug 25, 2023
377582f
Merge pull request #1073 from liuchangshiye/mlp-malaria-cnn
nudles Aug 28, 2023
d9831d9
Add the training file for model selection
NLGithubWP Aug 28, 2023
1dc8487
Merge pull request #1074 from NLGithubWP/add_train_mlp
chrishkchris Aug 29, 2023
ba5f573
Add SGD optimizer for model selection
NLGithubWP Aug 30, 2023
b56eaab
Merge pull request #1075 from NLGithubWP/update_train_mlp
chrishkchris Aug 31, 2023
53c664f
Add implementation for a single optimization step in model selection
NLGithubWP Aug 31, 2023
1e55cfb
Merge pull request #1076 from NLGithubWP/single_opt_ms
lzjpaul Sep 1, 2023
dc1a7f3
Increase the step counter, learning rate for the model selection SGD
liuchangshiye Sep 1, 2023
bb878cd
Merge pull request #1078 from liuchangshiye/model-selection-psql
lzjpaul Sep 4, 2023
7f0152a
Add implementation for get and set states
NLGithubWP Sep 2, 2023
af1634c
Merge pull request #1080 from NLGithubWP/update_train_mlp_
nudles Sep 5, 2023
96b0799
Add the implementation for model selection optimizer
daoducanhc Sep 5, 2023
5bdb8fb
Merge pull request #1082 from daoducanhc/chris-sep-5th
chrishkchris Sep 5, 2023
3f1d9cc
Add implementation for mlp models in model selection
Sep 5, 2023
d47766c
Merge pull request #1083 from zmeihui/23-9-5-ms
lzjpaul Sep 6, 2023
94e8e43
Add implementation for sum error loss for model selection
daoducanhc Sep 6, 2023
210d7a6
Merge pull request #1085 from daoducanhc/chris-sep-6th
chrishkchris Sep 6, 2023
0c151e7
Add the backward implementation for sum error loss
NLGithubWP Sep 6, 2023
ec2d81c
Merge pull request #1086 from NLGithubWP/add_model
lzjpaul Sep 6, 2023
2cf2a5b
Add the SumErrorLayer implementation
liuchangshiye Sep 6, 2023
e6cb95f
Merge pull request #1088 from liuchangshiye/model-selection-psql-msmlp
nudles Sep 7, 2023
05cd43c
Create the ms_model_mlp folder for dynamic model creation
Sep 7, 2023
915d7d9
Merge pull request #1089 from zmeihui/23-9-7-ms-model
lzjpaul Sep 7, 2023
d14fc3b
Add the autograd implementation for msmlp
daoducanhc Sep 7, 2023
c5f210e
Merge pull request #1090 from daoducanhc/chris-sep-7th
chrishkchris Sep 7, 2023
62a94cd
Add train one batch function for the dynamic model
KimballCai Sep 7, 2023
5b36459
Merge pull request #1091 from KimballCai/dev-postgresql
lzjpaul Sep 7, 2023
9e955b1
Add create model function for the dynamic model
liuchangshiye Sep 8, 2023
20c9d79
Merge pull request #1092 from liuchangshiye/model-selection-psql-ms-m…
lzjpaul Sep 8, 2023
f84ebe5
Add training process for the dynamic model
NLGithubWP Sep 8, 2023
5df0cec
Merge pull request #1093 from NLGithubWP/update_model
lzjpaul Sep 8, 2023
41e260d
Add autograd implementation for dynamic model creation
Sep 11, 2023
9d89392
Merge pull request #1094 from zmeihui/23-9-11-autograd
chrishkchris Sep 11, 2023
1d2f6e7
Debug the training process for training free model evaluation metric
NLGithubWP Sep 12, 2023
7154dad
Merge pull request #1095 from NLGithubWP/dev-postgresql
lzjpaul Sep 12, 2023
369f0f2
Training process for mlp models with varying layer sizes
lzjpaul Sep 14, 2023
cc741f5
Merge pull request #1096 from lzjpaul/23-9-14-ms
chrishkchris Sep 14, 2023
2c955ee
Training script for model selection
lzjpaul Sep 15, 2023
aa5cc42
Merge pull request #1097 from lzjpaul/23-9-15-ms
chrishkchris Sep 15, 2023
957caaf
Update the training file for dynamic models
Wu-Junran Sep 16, 2023
95a5536
Merge pull request #1099 from Wu-Junran/junran-dev-postgresql
lzjpaul Sep 16, 2023
593dc78
Add the doap file for the project
Sep 20, 2023
22252e7
Merge pull request #1100 from zmeihui/23-9-20-doap
lzjpaul Sep 20, 2023
41a3772
Add the training script for CNN models
daoducanhc Sep 22, 2023
931a312
Merge pull request #1101 from daoducanhc/chris-sep-22th
lzjpaul Sep 22, 2023
a7f59a1
Add the training script for models using MPI
NLGithubWP Sep 23, 2023
41aa437
Merge pull request #1102 from NLGithubWP/add_train_mpi
lzjpaul Sep 23, 2023
c7752fa
Model class file for dynamic models
NLGithubWP Oct 2, 2023
5c71962
Merge pull request #1103 from NLGithubWP/dynamic_models
lzjpaul Oct 2, 2023
5b8d928
Add the ModelMeta for dynamic models
lzjpaul Oct 3, 2023
647e5f3
Merge pull request #1104 from lzjpaul/23-10-3-modelmeta
chrishkchris Oct 3, 2023
26deb2c
Update the ModelMeta class
daoducanhc Oct 5, 2023
a73c7b2
Merge pull request #1105 from daoducanhc/chris-oct-5th
chrishkchris Oct 5, 2023
8f3a7ab
Add the running script for model selection
Oct 6, 2023
aa683ce
Merge pull request #1106 from zmeihui/23-10-6-run
lzjpaul Oct 6, 2023
1e4e272
Add the training script for model selection
NLGithubWP Oct 7, 2023
772ba55
Merge pull request #1107 from NLGithubWP/add_ms_model
lzjpaul Oct 7, 2023
3b9658f
Add data processing functions for model selection
lzjpaul Oct 8, 2023
30b7980
Merge pull request #1108 from lzjpaul/23-10-8-data-process
chrishkchris Oct 9, 2023
3ce7c33
Add the optimizer for model selection
Oct 9, 2023
9d8a2f5
Merge pull request #1109 from zmeihui/23-10-9-optimizer
lzjpaul Oct 10, 2023
063effd
Add the implementation of MSSGD for model selection
daoducanhc Oct 11, 2023
830e1f7
Merge pull request #1110 from daoducanhc/chris-oct-11th
lzjpaul Oct 11, 2023
619d119
Add the implementation of a single optimization step for model selection
NLGithubWP Oct 12, 2023
31ea937
Merge pull request #1111 from NLGithubWP/add_opt_ms
chrishkchris Oct 12, 2023
850bcf1
Add the implementation for the model selection example
lzjpaul Oct 24, 2023
b911ce2
Merge pull request #1112 from lzjpaul/23-10-24-v410
chrishkchris Oct 24, 2023
daa9c17
Update license headers for v4.1.0
lzjpaul Oct 25, 2023
3990365
Merge pull request #1113 from lzjpaul/23-10-24-v410-1
chrishkchris Oct 25, 2023
b25cd92
Add the release notes for v4.1.0
Oct 25, 2023
17ac0ad
Merge pull request #1114 from zmeihui/23-10-25-release-note
lzjpaul Oct 25, 2023
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -29,10 +29,10 @@ LIST(APPEND CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR}/cmake/Thirdparty)
#string(REGEX REPLACE "^[0-9]+\\.[0-9]+\\.([0-9]+).*" "\\1" VERSION_PATCH "${VERSION}")


SET(PACKAGE_VERSION 4.0.0) # ${VERSION})
SET(VERSION 4.0.0)
SET(PACKAGE_VERSION 4.1.0) # ${VERSION})
SET(VERSION 4.1.0)
SET(SINGA_MAJOR_VERSION 4)
SET(SINGA_MINOR_VERSION 0)
SET(SINGA_MINOR_VERSION 1)
SET(SINGA_PATCH_VERSION 0)
#SET(SINGA_MAJOR_VERSION ${VERSION_MAJOR}) # 0 -
#SET(SINGA_MINOR_VERSION ${VERSION_MINOR}) # 0 - 9
Expand Down
56 changes: 56 additions & 0 deletions LICENSE
Original file line number Diff line number Diff line change
Expand Up @@ -559,3 +559,59 @@ PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

===============================================================================
SINGA bundles the following under MIT License:
examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/pg_extension/*

MIT License

Portions Copyright 2019-2021 ZomboDB, LLC.
Portions Copyright 2021-2023 Technology Concepts & Design, Inc.
Portions Copyright 2023 PgCentral Foundation, Inc.

All rights reserved.

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.

===============================================================================
SINGA bundles the following under The PostgreSQL License:
examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/pg_extension/*
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

the same set of files are under MIT and PostgreSQL?
if these are written by SINGA contributor (i.e., not copied from other projects), they should just be released under apache license.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

these source files are written by us, so they should be released under apache license


The PostgreSQL License

Portions Copyright (c) 1996-2023, The PostgreSQL Global Development Group

Portions Copyright (c) 1994, The Regents of the University of California

Permission to use, copy, modify, and distribute this software and its documentation for any
purpose, without fee, and without a written agreement is hereby granted, provided that the above
copyright notice and this paragraph and the following two paragraphs appear in all copies.

IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT,
SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING LOST PROFITS, ARISING
OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF
THE UNIVERSITY OF CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING,
BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS ON AN "AS IS" BASIS,
AND THE UNIVERSITY OF CALIFORNIA HAS NO OBLIGATIONS TO PROVIDE MAINTENANCE, SUPPORT,
UPDATES, ENHANCEMENTS, OR MODIFICATIONS.

13 changes: 13 additions & 0 deletions NOTICE
Original file line number Diff line number Diff line change
Expand Up @@ -30,3 +30,16 @@ developers of Apache SINGA under Apache License, Version 2.0.
./doc/_static/images/sgd.png
./doc/_static/images/singa.png
./doc/_static/images/singav1-sw.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/documents/image-20231020174425377.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/documents/image-20231020174945226.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230421214835152.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230421220338391.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230421220443231.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230702035554579.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230702035622198.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230702035639502.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230702035806963.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230722202555763.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230722205244718.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230724111325368.png
./examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/documents/imgs/image-20230724111659545.png
37 changes: 37 additions & 0 deletions RELEASE_NOTES
Original file line number Diff line number Diff line change
@@ -1,3 +1,40 @@
Release Notes - SINGA - Version singa-4.1.0

SINGA is a distributed deep learning library.

This release includes following changes:

* New examples
* Add an example for malaria detection using cell images.
* Add an example for structured data learning.

* Add support for models running on top of RDBMS
* Add support for in-database model definition and selection in RDBMS.
* Implement training-free model evaluation metrics for in-database model selection.
* Implement a coordinator to balance between training-free and training-based model evaluations
for in-database model selection.

* Enhance distributed training
* Add implementations for the sum error loss.
* Improve the optimizer to return model gradients.
* Improve the iterative checking for tensors and strings in the ModelMeta class.

* Enhance example code
* Add support for flexible setting of training configurations for models, e.g., learning rates,
weight decay, momentum, etc.
* Add implementations for dynamic models with varying layer sizes.

* Update the website
* Add illustrations for database integration.
* Update users of Apache SINGA.

* Fix bugs
* Update the NVIDIA_GPGKEY in the Dockerfile for building wheel files.
* Update the versions of dependencies in the wheel file.
* Fix the collections module in the model.py file.

----------------------------------------------------------------------------------------------

Release Notes - SINGA - Version singa-4.0.0

SINGA is a distributed deep learning library.
Expand Down
63 changes: 63 additions & 0 deletions doap/doap_SINGA.rdf
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
<?xml version="1.0"?>
<?xml-stylesheet type="text/xsl"?>
<rdf:RDF xml:lang="en"
xmlns="http://usefulinc.com/ns/doap#"
xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:asfext="http://projects.apache.org/ns/asfext#"
xmlns:foaf="http://xmlns.com/foaf/0.1/">
<!--
Licensed to the Apache Software Foundation (ASF) under one or more
contributor license agreements. See the NOTICE file distributed with
this work for additional information regarding copyright ownership.
The ASF licenses this file to You under the Apache License, Version 2.0
(the "License"); you may not use this file except in compliance with
the License. You may obtain a copy of the License at

https://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
-->
<Project rdf:about="https://singa.apache.org/">
<created>2023-09-06</created>
<license rdf:resource="https://spdx.org/licenses/Apache-2.0" />
<name>Apache SINGA</name>
<homepage rdf:resource="https://singa.apache.org/" />
<asfext:pmc rdf:resource="https://singa.apache.org" />
<shortdesc>A Distributed Deep Learning Library</shortdesc>
<description>Apache SINGA is an Apache top-level project for developing an open-source machine learning library. It provides a flexible architecture for scalable distributed training, is extensible to run over a wide range of hardware, and has a focus on healthcare applications.</description>
<mailing-list rdf:resource="https://singa.apache.org/docs/mail-lists/" />
<download-page rdf:resource="https://singa.apache.org/docs/downloads/" />
<programming-language>C++</programming-language>
<category rdf:resource="https://projects.apache.org/category/big-data" />
<release>
<Version>
<name>Apache SINGA 4.0.0</name>
<created>2023-04-07</created>
<revision>4.0.0</revision>
</Version>
</release>
<repository>
<SVNRepository>
<location rdf:resource="https://dist.apache.org/repos/dist/dev/singa/"/>
<browse rdf:resource="https://dist.apache.org/repos/dist/release/singa/"/>
</SVNRepository>
</repository>
<repository>
<GitRepository>
<location rdf:resource="https://github.com/apache/singa"/>
<browse rdf:resource="https://github.com/apache/singa"/>
</GitRepository>
</repository>
<maintainer>
<foaf:Person>
<foaf:name>LUO ZHAOJING</foaf:name>
<foaf:mbox rdf:resource="mailto:[email protected]"/>
</foaf:Person>
</maintainer>
</Project>
</rdf:RDF>

24 changes: 24 additions & 0 deletions examples/armnet/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
<!--
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing,
software distributed under the License is distributed on an
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
-->

## ARM-Net: Adaptive Relation Modeling Network for Structured Data

![version](https://img.shields.io/badge/version-v3.5-green)
![python](https://img.shields.io/badge/python-3.8.3-blue)
![singa](https://img.shields.io/badge/singa-3.1.0-orange)

This folder contains our Singa implementation of [ARM-Net: Adaptive Relation Modeling Network for Structured Data](https://dl.acm.org/doi/10.1145/3448016.3457321).
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The folder does not have the source code files..

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

For this release, we will remove this part

4 changes: 4 additions & 0 deletions examples/cifar_distributed_cnn/run-rtx.sh
Original file line number Diff line number Diff line change
Expand Up @@ -38,3 +38,7 @@ mpiexec -np 8 python train_mpi.py mlp cifar100 -l 0.015 -b 32
mpiexec -np 8 python train_mpi.py alexnet mnist -l 0.015 -b 32
mpiexec -np 8 python train_mpi.py alexnet cifar10 -l 0.015 -b 32
mpiexec -np 8 python train_mpi.py alexnet cifar100 -l 0.015 -b 32

# xceptionnet
mpiexec -np 8 python train_mpi.py xceptionnet mnist -l 0.015 -b 32
mpiexec -np 8 python train_mpi.py xceptionnet cifar10 -l 0.015 -b 32
2 changes: 2 additions & 0 deletions examples/cnn/run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@
# under the License.
#

#!/usr/bin/env python -W ignore::DeprecationWarning

### mnist
python train_cnn.py mlp mnist
python train_cnn.py cnn mnist
Expand Down
44 changes: 44 additions & 0 deletions examples/malaria_cnn/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
<!--
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing,
software distributed under the License is distributed on an
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
-->

# Singa for Malaria Detection Task

## Malaria

Malaria is caused by parasites and could be transmitted through infected mosquitoes. There are about 200 million cases worldwide, and about 400,000 deaths per year, therefore, malaria does lots of harm to global health.

Although Malaria is a curable disease, inadequate diagnostics make it harder to reduce mortality, as a result, a fast and reliable diagnostic test is a promising and effective way to fight malaria.

To mitigate the problem, we use Singa to implement a machine learning model to help with Malaria diagnosis. The dataset is from Kaggle https://www.kaggle.com/datasets/miracle9to9/files1?resource=download. Please download the dataset before running the scripts.

## Structure

* `data` includes the scripts for preprocessing Malaria image datasets.

* `model` includes the CNN model construction codes by creating
a subclass of `Module` to wrap the neural network operations
of each model.

* `train_cnn.py` is the training script, which controls the training flow by
doing BackPropagation and SGD update.

## Command
```bash
python train_cnn.py cnn malaria -dir pathToDataset
```
Loading
Loading