Compare commits
5 Commits
66727770d8
...
main
Author | SHA1 | Date | |
---|---|---|---|
8a1442995b | |||
9fcd0786b1 | |||
bc504fce74 | |||
eb23c35b0f | |||
0da0e66710 |
4
.gitignore
vendored
4
.gitignore
vendored
@ -1,4 +1,6 @@
|
||||
nsmc/**/*
|
||||
nsmc.zip
|
||||
.ipynb_checkpoints/**/*
|
||||
__pycache__/**/*
|
||||
__pycache__/**/*
|
||||
model.zip
|
||||
model/**/*
|
14
Batch.ipynb
14
Batch.ipynb
@ -3,7 +3,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 1,
|
||||
"id": "c916dd3b",
|
||||
"id": "5a4a1e30",
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
@ -25,7 +25,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 2,
|
||||
"id": "d5861234",
|
||||
"id": "710cd5b2",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
@ -39,7 +39,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 3,
|
||||
"id": "5accd3a9",
|
||||
"id": "da018ffe",
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
@ -68,7 +68,7 @@
|
||||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "d10fcb83",
|
||||
"id": "69f05cf6",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"data를 준비"
|
||||
@ -77,7 +77,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 7,
|
||||
"id": "552fe555",
|
||||
"id": "961edd10",
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
@ -114,7 +114,7 @@
|
||||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "1cff8e03",
|
||||
"id": "4178b576",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"간단한 collate function"
|
||||
@ -123,7 +123,7 @@
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "89eb64d8",
|
||||
"id": "a5ff0049",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": []
|
||||
|
16
LICENSE
Normal file
16
LICENSE
Normal file
@ -0,0 +1,16 @@
|
||||
MIT License
|
||||
|
||||
Copyright 2022 monoid
|
||||
|
||||
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.
|
||||
|
||||
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.
|
2304
Training.ipynb
Normal file
2304
Training.ipynb
Normal file
File diff suppressed because one or more lines are too long
@ -7,13 +7,15 @@ from ndata import readNsmcRawData, NsmcRawData
|
||||
|
||||
def readNsmcDataAll():
|
||||
"""
|
||||
Returns: train, test
|
||||
Returns: train, dev, test
|
||||
"""
|
||||
print("read train set", file=sys.stderr)
|
||||
train = readNsmcRawData("nsmc/nsmc-master/ratings_train.txt",use_tqdm=True,total=150_000)
|
||||
print("read test set", file=sys.stderr)
|
||||
test = readNsmcRawData("nsmc/nsmc-master/ratings_test.txt",use_tqdm=True,total=50_000)
|
||||
return NsmcDataset(train),NsmcDataset(test)
|
||||
testBig = readNsmcRawData("nsmc/nsmc-master/ratings_test.txt",use_tqdm=True,total=50_000)
|
||||
test = testBig[:30_000]
|
||||
dev = testBig[30_000:]
|
||||
return NsmcDataset(train),NsmcDataset(dev),NsmcDataset(test)
|
||||
|
||||
class NsmcDataset(Dataset):
|
||||
def __init__(self, data: List[NsmcRawData]):
|
||||
|
Loading…
Reference in New Issue
Block a user